Add Llama 4 to FA3 test (#5509)

2025-04-26 19:49:31 -07:00
parent 094891c01a
commit 408ba02218
4 changed files with 188 additions and 114 deletions
--- a/test/srt/run_suite.py
+++ b/test/srt/run_suite.py
@@ -30,7 +30,6 @@ suites = {
        TestFile("test_chunked_prefill.py", 336),
        TestFile("test_eagle_infer.py", 500),
        TestFile("test_ebnf_constrained.py"),
-        TestFile("test_fa3.py", 200),
        TestFile("test_fp8_kernel.py", 8),
        TestFile("test_embedding_openai_server.py", 36),
        TestFile("test_hidden_states.py", 55),
@@ -91,6 +90,9 @@ suites = {
        TestFile("test_update_weights_from_distributed.py", 100),
        TestFile("test_verl_engine.py", 100),
    ],
+    "per-commit-8-gpu": [
+        TestFile("test_fa3.py", 30),
+    ],
    "nightly": [
        TestFile("test_nightly_gsm8k_eval.py"),
    ],