[MPS] Add scatter gather kernels (support up to 5 dimensions) #94663
Conversation
🔗 Helpful Links🧪 See artifacts and rendered test results at hud.pytorch.org/pr/94663
Note: Links to docs will display an error until the docs builds have been completed. ✅ No FailuresAs of commit d73c1da: This comment was automatically generated by Dr. CI and updates every 15 minutes. |
* Add scatter gather kernels (support up to 5 dimensions) * Remove duplicate code * Return gather output directly if dst is contiguous * Fix cast gather scatter test * Address PR comments * Address PR comments
DenisVieriu97
force-pushed
the
scatter_gather_kernels_5d
branch
from
88d2da7
to
31e86ae
Compare
Merge startedYour change will be merged once all checks pass (ETA 0-4 Hours). Learn more about merging in the wiki. Questions? Feedback? Please reach out to the PyTorch DevX Team |
|
The merge job was canceled. If you believe this is a mistake,then you can re trigger it through pytorch-bot. |
Merge startedYour change will be merged once all checks pass (ETA 0-4 Hours). Learn more about merging in the wiki. Questions? Feedback? Please reach out to the PyTorch DevX Team |
Merge failedReason: New commits were pushed while merging. Please rerun the merge command. Details for Dev Infra teamRaised by workflow job |
|
@pytorchbot merge -g |
Merge startedYour change will be merged once all checks pass (ETA 0-4 Hours). Learn more about merging in the wiki. Questions? Feedback? Please reach out to the PyTorch DevX Team |
…pytorch#94663)" The new scatter gather kernels result in ~5% reduced performance during inference with Stable Diffusion.
…pytorch#94663)" This reverts commit b794fd1.
…pytorch#94663)" This reverts commit b794fd1.
…pytorch#94663)" This reverts commit b794fd1.
Add scatter gather kernels (support up to 5 dimensions)
mH,mT,T,Hon Monterey