Skip to content

[SYCL free func] Migrate softmax kernels to sycl free functions - #5394

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

YzyParry wants to merge 1 commit into
intel:mainfrom
YzyParry:SoftMaxKernels

Conversation

@YzyParry

Copy link
Copy Markdown
Contributor

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

  • DispatchSoftmaxForwardKernel
  • DispatchSoftmaxBackwardKernel
  • SoftmaxForwardKernelFunctor
  • SpatialSoftmaxForwardKernelFunctor
  • SoftmaxBackwardKernelFunctor
  • SpatialSoftmaxBackwardKernelFunctor

@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 Sep 16, 2026
* refactor DispatchSoftmaxForwardKernel by free function

* refactor DispatchSoftmaxBackwardKernel by free function

* refactor SoftmaxForwardKernelFunctor by free function

* refactor SpatialSoftmaxForwardKernelFunctor by free function

* refactor SoftmaxBackwardKernelFunctor by free function

* refactor SpatialSoftmaxBackwardKernelFunctor by free function
@YzyParry
YzyParry marked this pull request as ready for review September 20, 2026 06:01
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