diff --git a/op_tests/triton_tests/test_la.py b/op_tests/triton_tests/test_la.py index b4469e56a3..946e2c6e97 100644 --- a/op_tests/triton_tests/test_la.py +++ b/op_tests/triton_tests/test_la.py @@ -11,6 +11,7 @@ ) from aiter.ops.triton._triton_kernels.lean_atten import _get_config import aiter.ops.triton.utils._triton.arch_info as arch_info +import pytest def get_lean_attn_inputs( @@ -324,6 +325,7 @@ def reference_attention(q, k, v, n_ctx, n_ctx_q, sm_scale, causal): # (False, 1, 64, 4096, [4096], 128, 304, torch.float16, 128, 16, 3, 4), ], ) +@pytest.mark.skip(reason="This test is temporarily disabled.") def test_persistent_lean_attention( request, causal,