Skip to content

Commit d4d41f9

Browse files
authored
fix skip condition for test_sdp_bias_bc (#5112) (#5113)
1 parent 05be170 commit d4d41f9

File tree

1 file changed

+8
-1
lines changed

1 file changed

+8
-1
lines changed

tests/gpu/examples/test_sdp_bias_bc.py

Lines changed: 8 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -8,7 +8,14 @@
88

99

1010
class TestTorchMethod(TestCase):
11-
@pytest.mark.skipif(not torch.xpu.has_xetla(), reason="fallback is required")
11+
@pytest.mark.skipif(
12+
not (
13+
torch.xpu.has_xetla()
14+
and torch.xpu.has_2d_block_array()
15+
and torch.xpu.has_xmx()
16+
),
17+
reason="fallback is required",
18+
)
1219
def test_sdp_bias_bc(self, dtype=torch.bfloat16):
1320
b = 2
1421
n = 2

0 commit comments

Comments
 (0)