Skip to content

[SYCL free func] Migrate sorting kernels to sycl free function - #5130

Open
YzyParry wants to merge 1 commit into
intel:mainfrom
YzyParry:Sorting
Open

[SYCL free func] Migrate sorting kernels to sycl free function#5130
YzyParry wants to merge 1 commit into
intel:mainfrom
YzyParry:Sorting

Conversation

@YzyParry

@YzyParry YzyParry commented Aug 28, 2026

Copy link
Copy Markdown

Migrate kernels in src/ATen/native/xpu/sycl/Sorting.cpp to sycl free function.
Including:

  • GatherMedianKernelFunctor
  • GatherKthValueKernelFunctor

src/ATen/native/xpu/sycl/SortingRadixSelect.h is updated accordingly. The device helpers countRadixUsingMask, findPattern and radixSelect now take a raw int* smem instead of const sycl_local_acc_t<int>&, so the free function kernels can pass the pointer returned by syclexp::get_work_group_scratch_memory().

@github-actions github-actions Bot added disable_e2e Disable all e2e test jobs for the PR disable_distributed Disable distributed UT test jobs for the PR labels Aug 28, 2026
@YzyParry
YzyParry marked this pull request as ready for review August 28, 2026 03:02
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

disable_distributed Disable distributed UT test jobs for the PR disable_e2e Disable all e2e test jobs for the PR

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant