From 9cd9dc83b31e70723fb28ad820994a446cd57f8b Mon Sep 17 00:00:00 2001 From: Lianmin Zheng Date: Mon, 16 Dec 2024 14:17:27 -0800 Subject: [PATCH] Temporarily disable unit test of torch native attention backend (#2492) --- test/srt/run_suite.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/test/srt/run_suite.py b/test/srt/run_suite.py index 89fdacfa2..137507656 100644 --- 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",