Skip to content

[Graph| tests, example, doc] Add GQA v2 support for implicit causal mask and example, doc update #3409

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.

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

Open
wants to merge 4 commits into
base: main
Choose a base branch
from

Conversation

xiang1guo
Copy link
Contributor

@xiang1guo xiang1guo commented Jun 10, 2025

Follow-up work based on #2931

  • GQA v2 implicit causal mask test cases support
  • Add example for GQA v2 pattern.
  • Doc update.

@xiang1guo xiang1guo self-assigned this Jun 10, 2025
@xiang1guo xiang1guo requested review from a team as code owners June 10, 2025 08:22
@xiang1guo xiang1guo added the component:graph-api Codeowner: @oneapi-src/onednn-graph label Jun 10, 2025
@github-actions github-actions bot added component:tests Codeowner: @oneapi-src/onednn-arch component:examples labels Jun 10, 2025
@xiang1guo xiang1guo requested a review from a team as a code owner June 11, 2025 02:52
@github-actions github-actions bot added the documentation A request to change/fix/improve the documentation. Codeowner: @oneapi-src/onednn-doc label Jun 11, 2025
@xiang1guo xiang1guo changed the title [Graph, tests, example] Add GQA v2 support for implicit causal mask and example [Graph| tests, example, doc] Add GQA v2 support for implicit causal mask and example, doc update Jun 11, 2025
@xiang1guo xiang1guo force-pushed the xiang/main/gqa-follow-up branch from 29563cf to 80aeeff Compare June 11, 2025 02:56
@xiang1guo xiang1guo force-pushed the xiang/main/gqa-follow-up branch from c299d16 to 21779d1 Compare June 16, 2025 00:44
@xiang1guo
Copy link
Contributor Author

make test
set test_scope=NIGHTLY
disable benchdnn_all
enable benchdnn_graph

@xiang1guo xiang1guo force-pushed the xiang/main/gqa-follow-up branch from 21779d1 to 4870e0f Compare June 17, 2025 01:14
@xiang1guo
Copy link
Contributor Author

make test
set test_scope=NIGHTLY
disable benchdnn_all
enable benchdnn_graph

@xiang1guo xiang1guo force-pushed the xiang/main/gqa-follow-up branch from 4870e0f to dee686e Compare June 17, 2025 06:22
@xiang1guo xiang1guo force-pushed the xiang/main/gqa-follow-up branch from dee686e to 8456adf Compare June 17, 2025 09:15
@xiang1guo
Copy link
Contributor Author

make test
set test_scope=NIGHTLY
disable benchdnn_all
enable benchdnn_graph

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
component:examples component:graph-api Codeowner: @oneapi-src/onednn-graph component:tests Codeowner: @oneapi-src/onednn-arch documentation A request to change/fix/improve the documentation. Codeowner: @oneapi-src/onednn-doc
Projects
None yet
Development

Successfully merging this pull request may close these issues.

5 participants