Temporarily disable unit test of torch native attention backend (#2492)

2024-12-16 14:17:27 -08:00
parent 7a1aecb938
commit 9cd9dc83b3
1 changed files with 2 additions and 1 deletions
--- a/test/srt/run_suite.py
+++ b/test/srt/run_suite.py
@@ -34,7 +34,8 @@ suites = {
        "test_srt_endpoint.py",
        "test_torch_compile.py",
        "test_torch_compile_moe.py",
-        "test_torch_native_attention_backend.py",
+        # Temporarily disable this because it requires PyTorch >= 2.5
+        # "test_torch_native_attention_backend.py",
        "test_torchao.py",
        "test_triton_attention_kernels.py",
        "test_triton_attention_backend.py",