[Fix] Fix and rename flashmla CI test (#6045)

2025-05-06 13:25:15 -07:00
parent c9abd7be01
commit bdd17998e6
3 changed files with 87 additions and 65 deletions
--- a/test/srt/run_suite.py
+++ b/test/srt/run_suite.py
@@ -51,7 +51,7 @@ suites = {
        TestFile("test_mla_int8_deepseek_v3.py", 389),
        TestFile("test_mla_flashinfer.py", 395),
        TestFile("test_mla_fp8.py", 153),
-        TestFile("test_flash_mla_attention_backend.py", 300),
+        TestFile("test_flashmla.py", 300),
        TestFile("test_no_chunked_prefill.py", 108),
        TestFile("test_no_overlap_scheduler.py", 216),
        TestFile("test_openai_server.py", 149),