Skip to content

Commit d5177f3

Browse files
iupaikov-amdjithunnair-amd
authored andcommitted
Skipped one sdpa test for unsupported platforms
1 parent 15ba5b5 commit d5177f3

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

test/inductor/test_flex_decoding.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1332,6 +1332,7 @@ def mask_mod(b, h, q, kv):
13321332
self.assertEqual(query.grad[:, :, M:, :].sum(), 0)
13331333

13341334
@supported_platform
1335+
@skipIf(not PLATFORM_SUPPORTS_FLASH_ATTENTION, "Some archs don't support SDPA")
13351336
def test_windowed_no_mask_vs_sdpa(self):
13361337
score_mod = _generate_windowed(1000)
13371338
attention = functools.partial(flex_attention, score_mod=score_mod)

0 commit comments

Comments
 (0)