fix b200 ci (#14786)

This commit is contained in:
b8zhong
2025-12-09 23:08:41 -08:00
committed by GitHub
parent 793c98afaf
commit b0a25d0913
2 changed files with 3 additions and 3 deletions

View File

@@ -171,8 +171,8 @@ suites = {
TestFile("test_deepseek_v3_fp4_4gpu.py", 1800),
TestFile("test_flash_attention_4.py", 300),
TestFile("test_fp8_blockwise_gemm.py", 280),
TestFile("test_gpt_oss_4gpu.py", 600),
TestFile("test_llama31_fp4.py", 300),
TestFile("test_gpt_oss_4gpu.py", 700),
TestFile("test_llama31_fp4.py", 90),
TestFile("test_eagle_infer_beta_dp_attention.py", 300),
],
# "per-commit-8-gpu-b200": [

View File

@@ -49,7 +49,7 @@ class TestLlama31FP4(unittest.TestCase):
metrics = run_eval_few_shot_gsm8k(args)
print(metrics)
self.assertGreater(metrics["accuracy"], 0.54)
self.assertGreater(metrics["accuracy"], 0.64)
if __name__ == "__main__":