Skip to content

Navigation Menu

Appearance settings

Explore
By company size
By use case
By industry
View all solutions
Topics
- AI
- DevOps
- Security
- Software Development
- View all
Explore
- GitHub Sponsors
  Fund open source developers
- The ReadME Project
  GitHub community articles
Repositories
- Enterprise platform
  AI-powered developer platform
Available add-ons
Pricing

Search code, repositories, users, issues, pull requests...

Search

Clear

Search syntax tips

Provide feedback

We read every piece of feedback, and take your input very seriously.

Include my email address so I can be contacted

Saved searches

Use saved searches to filter your results more quickly

Name

Query

To see all available qualifiers, see our documentation.

Appearance settings

You signed in with another tab or window. Reload to refresh your session. You signed out in another tab or window. Reload to refresh your session. You switched accounts on another tab or window. Reload to refresh your session.

Dismiss alert

uxlfoundation / oneDNN Public

Notifications You must be signed in to change notification settings
Fork 1k
Star 3.8k

Code
Issues 33
Pull requests 121
Actions
Wiki
Security
Insights

Additional navigation options

Code
Issues
Pull requests
Actions
Wiki
Security
Insights

graph: doc, interface, backend: support SDPA training#3396

Open

ElaineBao wants to merge 16 commits intomainuxlfoundation/oneDNN:mainfrom

yixin/sdpa-training-impluxlfoundation/oneDNN:yixin/sdpa-training-implCopy head branch name to clipboard

Conversation Commits16 (16)Checks Files changed

Open

graph: doc, interface, backend: support SDPA training#3396
ElaineBao wants to merge 16 commits intomainfrom
yixin/sdpa-training-impl

Commits

Commits on Jun 19, 2025

doc: graph: support new dtype combination for SoftMaxBackward
ElaineBao
committed
graph: interface: support new dtype combination for SoftMaxBackward
ElaineBao
committed
graph: backend: dnnl: support new dtype combination for SoftMaxBackward
ElaineBao
committed
benchdnn: inputs: graph: add cases for SoftMaxBackward
ElaineBao
committed
graph: interface: add optional stats output to SoftMax op
ElaineBao
committed
graph: backend: dnnl: add optional stats output to SoftMax op
ElaineBao
committed
graph: backend: dnnl: support sdpa training backward pattern
ElaineBao
committed
examples: graph: add sdpa training example
ElaineBao
committed
doc: graph: add optional stats output to SoftMax op
ElaineBao
committed
tests: benchdnn: graph: support custom driver of softmax with stats
ElaineBao
committed
graph: backend: dnnl: support sdpa training forward with large partition
ElaineBao
committed
tests: benchdnn: graph: recomputing softmax stats for input displacer
ElaineBao
committed
tests: benchdnn: graph: adjust correctness check for sdpa training
ElaineBao
committed
examples: graph: update sdpa training example
ElaineBao
committed
tests: benchdnn: graph: add cases for sdpa training
ElaineBao
committed
doc: graph: support SDPA training
ElaineBao
committed

Footer

© 2025 GitHub, Inc.

Footer navigation

Terms
Privacy
Security
Status
Docs
Contact

You can’t perform that action at this time.