fix b200 ci (#14786)
This commit is contained in:
@@ -171,8 +171,8 @@ suites = {
|
|||||||
TestFile("test_deepseek_v3_fp4_4gpu.py", 1800),
|
TestFile("test_deepseek_v3_fp4_4gpu.py", 1800),
|
||||||
TestFile("test_flash_attention_4.py", 300),
|
TestFile("test_flash_attention_4.py", 300),
|
||||||
TestFile("test_fp8_blockwise_gemm.py", 280),
|
TestFile("test_fp8_blockwise_gemm.py", 280),
|
||||||
TestFile("test_gpt_oss_4gpu.py", 600),
|
TestFile("test_gpt_oss_4gpu.py", 700),
|
||||||
TestFile("test_llama31_fp4.py", 300),
|
TestFile("test_llama31_fp4.py", 90),
|
||||||
TestFile("test_eagle_infer_beta_dp_attention.py", 300),
|
TestFile("test_eagle_infer_beta_dp_attention.py", 300),
|
||||||
],
|
],
|
||||||
# "per-commit-8-gpu-b200": [
|
# "per-commit-8-gpu-b200": [
|
||||||
|
|||||||
@@ -49,7 +49,7 @@ class TestLlama31FP4(unittest.TestCase):
|
|||||||
metrics = run_eval_few_shot_gsm8k(args)
|
metrics = run_eval_few_shot_gsm8k(args)
|
||||||
print(metrics)
|
print(metrics)
|
||||||
|
|
||||||
self.assertGreater(metrics["accuracy"], 0.54)
|
self.assertGreater(metrics["accuracy"], 0.64)
|
||||||
|
|
||||||
|
|
||||||
if __name__ == "__main__":
|
if __name__ == "__main__":
|
||||||
|
|||||||
Reference in New Issue
Block a user