Import flash_mla from sgl-kernel (#12135)

2025-10-29 23:54:21 -07:00
parent fb52d35f63
commit 621dfb8886
8 changed files with 9 additions and 63 deletions
@@ -82,7 +82,7 @@ suites = {
        TestFile("test_ebnf_constrained.py", 108),
        TestFile("test_eval_fp8_accuracy.py", 303),
        TestFile("test_fa3.py", 376),
-        # TestFile("test_flashmla.py", 352),
+        TestFile("test_flashmla.py", 352),
        TestFile("rotary_embedding/test_mrope.py", 300),
        TestFile("test_function_call_parser.py", 10),
        TestFile("test_fused_moe.py", 30),