Skip to content

Commit 9f390e6

Browse files
authored
Updated test_flex_decoding.py (#1947)
Fixes [SWDEV-480212](https://ontrack-internal.amd.com/browse/SWDEV-480212) test/inductor/test_flex_decoding.py runs fine and re-enable test/inductor/test_flex_decoding.py::TestFlexDecoding::test_windowed_no_mask_vs_sdpa (verified this unit test runs fine on NAVI3x and MI300)
1 parent a7d214b commit 9f390e6

File tree

1 file changed

+0
-1
lines changed

1 file changed

+0
-1
lines changed

test/inductor/test_flex_decoding.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1383,7 +1383,6 @@ def mask_mod(b, h, q, kv):
13831383
loss.backward()
13841384
self.assertEqual(query.grad[:, :, M:, :].sum(), 0)
13851385

1386-
@skipIfRocm
13871386
@supported_platform
13881387
def test_windowed_no_mask_vs_sdpa(self):
13891388
score_mod = _generate_windowed(1000)

0 commit comments

Comments
 (0)