8000 [Intel GPU] Add SDPA implementation on XPU with OneDNN by DDEle · Pull Request #147612 · pytorch/pytorch · GitHub
[go: up one dir, main page]
Skip to content

[Intel GPU] Add SDPA implementation on XPU with OneDNN #147612

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom
Closed
wants to merge 16 commits into from
Closed

[Intel GPU] Add SDPA implementation on XPU with OneDNN #147612

wants to merge 16 commits into from

Conversation

DDEle
Copy link
Contributor
@DDEle DDEle commented Feb 21, 2025 •
edited
Loading

Add XPU implementation of OneDNN based SDPA operator. Will be integrated and enabled later.

Depends on BUILD_GRAPH switch in #147608

cc @jgong5 @mingfeima @XiaobingSuper @sanchitintel @ashokei @jingxu10

@pytorch-bot PyTorch Bot
Copy link
pytorch-bot bot commented Feb 21, 2025 •
edited
Loading

🔗 Helpful Links

🧪 See artifacts and rendered test results at hud.pytorch.org/pr/147612

Note: Links to docs will display an error until the docs builds have been completed.

⏳ No Failures, 1 Pending

As of commit b20c410 with merge base bea7218 (image):
💚 Looks good so far! There are no failures yet. 💚

This comment was automatically generated by Dr. CI and updates every 15 minutes.

@pytorch-bot pytorch-bot bot added the module: cpu CPU specific problem (e.g., perf, algorithm) label Feb 21, 2025
@DDEle DDEle mentioned this pull request Feb 21, 2025
Closed
@DDEle
Copy link
Contributor Author
DDEle commented Feb 21, 2025

@pytorchbot label "topic: not user facing"