feat: add gpt oss b200 ci (#9988)

This commit is contained in:
Yineng Zhang
2025-09-03 17:26:38 -07:00
committed by GitHub
parent 397448ebbc
commit de9217334b
3 changed files with 5 additions and 10 deletions

View File

@@ -139,6 +139,7 @@ suites = {
],
"per-commit-8-gpu-b200": [
# add more here
TestFile("test_gpt_oss_4gpu.py", 600),
],
"per-commit-4-gpu-deepep": [
TestFile("ep/test_deepep_small.py", 531),

View File

@@ -9,10 +9,7 @@ class TestGptOss4Gpu(BaseTestGptOss):
model_variant="120b",
quantization="bf16",
expected_score_of_reasoning_effort={
"low": 0.61,
# remove to speed up
# "medium": 0.61,
# "high": 0.61,
"low": 0.60,
},
other_args=["--tp", "4", "--cuda-graph-max-bs", "200"],
)
@@ -22,10 +19,7 @@ class TestGptOss4Gpu(BaseTestGptOss):
model_variant="120b",
quantization="mxfp4",
expected_score_of_reasoning_effort={
"low": 0.61,
# remove to speed up
# "medium": 0.61,
# "high": 0.61,
"low": 0.60,
},
other_args=[
"--tp",