Skip to content

Commit fc74370

Browse files
authored
Update test_inductor_sdp.py (#4737)(#4721)
1 parent abc2ffb commit fc74370

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

tests/gpu/examples/test_inductor_sdp.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -31,6 +31,9 @@ class TestTorchMethod(TestCase):
3131
@pytest.mark.skipif(
3232
not torch.xpu.has_2d_block_array(), reason="ipex build without xetla"
3333
)
34+
@pytest.mark.skipif(
35+
not pytest.importorskip("triton"), reason="triton is not available"
36+
)
3437
def test_llama_sdp_fusion(self, dtype=torch.float16):
3538
query_states = torch.randn(
3639
(b, q, n, h), device=torch.device("xpu"), dtype=dtype, requires_grad=False

0 commit comments

Comments
 (0)