Skip to content

Commit 1fef591

Browse files
authored
[release/2.5] Added PLATFORM_SUPPORTS_MEM_EFF_ATTENTION skip logic for NAVI (#2393)
Fixes [SWDEV-510889](https://ontrack-internal.amd.com/browse/SWDEV-510889)
1 parent 33316c2 commit 1fef591

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

test/inductor/test_torchinductor.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -10282,6 +10282,9 @@ def fn(q, k, v):
1028210282
)
1028310283

1028410284
@expectedFailureXPU
10285+
@unittest.skipIf(
10286+
not PLATFORM_SUPPORTS_MEM_EFF_ATTENTION, "Some archs don't support mem eff SDPA"
10287+
)
1028510288
def test_scaled_dot_product_efficient_attention(self):
1028610289
if self.device == "cpu":
1028710290
raise unittest.SkipTest(f"requires {GPU_TYPE}")

0 commit comments

Comments
 (0)