Re-enable Flashinfer TRTLLM GEN MHA and Add Unit Test (#12885)

2025-11-10 20:17:43 -08:00
parent 9caca6a45c
commit 3594815a8b
3 changed files with 65 additions and 1 deletions
@@ -221,6 +221,7 @@ suites = {
    "nightly-4-gpu-b200": [
        TestFile("test_fp4_moe.py", 300),
        TestFile("nightly/test_gpt_oss_4gpu_perf.py", 600),
+        TestFile("nightly/test_flashinfer_trtllm_gen_attn_backend.py", 300),
    ],
    "nightly-8-gpu-b200": [],
    "nightly-4-gpu": [],