Skip to content

Commit 0bb3060

Browse files
committed
tests: benchdnn: graph: add case for sdpa
with non-contiguous strides in mask add
1 parent 09f095a commit 0bb3060

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

tests/benchdnn/inputs/graph/complex_fusion/harness_mha_all

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -91,6 +91,9 @@
9191
--reset --in-shapes=0:1x8x2x1x256+1:1x8x1x256x257+8:1x8x2x1x257+12:1x8x1x257x256 --case=complex_fusion/mha/gemma2-bf16-f32.json
9292
--reset --op-attrs=0:transpose_b:1 --case=complex_fusion/mha/gemma2-bf16-f32.json
9393

94+
# sdpa with non-contiguous strides in mask add
95+
--reset --in-shapes=5:1x1x384x384*147840x147840x385x1 --case=complex_fusion/mha/sdpa-plain-simplified-f16-f32.json
96+
9497
# Re-written graphs
9598
--reset --in-shapes=4:4x16x32x256+5:4x16x256x33+0:4x16x33x256+1:4x1x1x33+3:4x1x32x33 --case=complex_fusion/mha/MHA-GPT-inf-fp32-bs1.json
9699
--reset --mb=20 --case=complex_fusion/mha/MHA-bert_large-inf-fp32-bs1.json

0 commit comments

Comments
 (0)