Skip to content

Commit 589df12

Browse files
committed
benchdnn: inputs: graph: add sdpa regression tests
1 parent a157fbe commit 589df12

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

+3
Original file line numberDiff line numberDiff line change
@@ -113,3 +113,6 @@
113113
# d_qk != d_v
114114
--reset --dt=f32,bf16,f16 --in-shapes=8:1x16x384x32,8:1x16x384x64,8:1x16x384x128 --case=complex_fusion/mha/sdpa-plain-simplified-f16.json
115115
--reset --dt=f32,bf16,f16 --in-shapes=11:1x16x384x32,11:1x16x384x64,11:1x16x384x128 --case=complex_fusion/mha/sdpa-plain-implicit-causal-mask-fp32-bs1.json
116+
117+
# regression tests
118+
--reset --dt=f32,bf16,f16 --in-shapes=0:1x8x4096x16+1:1x8x5x16+5:1x1x4096x5+8:1x8x5x16 --case=complex_fusion/mha/sdpa-plain-simplified-f16.json

0 commit comments

Comments
 (0)