Skip to content

benchdnn: inputs: graph: add torchbench sdpa training tests#5050

Open
TaoLv wants to merge 3 commits intomainfrom
lvtao/main/torchbench-cases
Open

benchdnn: inputs: graph: add torchbench sdpa training tests#5050
TaoLv wants to merge 3 commits intomainfrom
lvtao/main/torchbench-cases

Conversation

@TaoLv
Copy link
Copy Markdown
Contributor

@TaoLv TaoLv commented Apr 20, 2026

Add more SDPA/GQA training test cases generated from PyTorch workloads and tests.

  • torchbench: 16 test cases over 6 new json files.
  • huggingface: 47 test cases over 6 new json files.
  • timms: 42 test cases over 2 new json files.

Fixes MFDNN-14845

@TaoLv TaoLv requested review from a team as code owners April 20, 2026 03:18
@github-actions github-actions Bot added the component:tests Codeowner: @oneapi-src/onednn-arch label Apr 20, 2026
@TaoLv
Copy link
Copy Markdown
Contributor Author

TaoLv commented Apr 20, 2026

make test
disable benchdnn_all
enable benchdnn_graph
set test_scope=NIGHTLY

@TaoLv TaoLv force-pushed the lvtao/main/torchbench-cases branch from 71ab132 to a83bf63 Compare April 27, 2026 07:56
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

component:tests Codeowner: @oneapi-src/onednn-arch

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants