[CI] Add Llama 3.1 8B FP4 to B200 CI (#12182)

2025-10-29 14:56:41 -07:00
parent df08f346ce
commit 7ed8ba05cb
2 changed files with 60 additions and 1 deletions
@@ -180,9 +180,10 @@ suites = {
        TestFile("test_disaggregation_pp.py", 140),
    ],
    "per-commit-4-gpu-b200": [
+        TestFile("test_deepseek_v3_fp4_4gpu.py", 3600),
        TestFile("test_flash_attention_4.py", 300),
        TestFile("test_gpt_oss_4gpu.py", 600),
-        TestFile("test_deepseek_v3_fp4_4gpu.py", 3600),
+        TestFile("test_llama31_fp4.py", 300),
    ],
    "per-commit-4-gpu-deepep": [
        TestFile("ep/test_deepep_small.py", 531),