Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
change
sglang
Commits
de921733
Unverified
Commit
de921733
authored
Sep 03, 2025
by
Yineng Zhang
Committed by
GitHub
Sep 03, 2025
Browse files
feat: add gpt oss b200 ci (#9988)
parent
397448eb
Changes
3
Show whitespace changes
Inline
Side-by-side
Showing
3 changed files
with
5 additions
and
10 deletions
+5
-10
scripts/ci/ci_install_dependency.sh
scripts/ci/ci_install_dependency.sh
+2
-2
test/srt/run_suite.py
test/srt/run_suite.py
+1
-0
test/srt/test_gpt_oss_4gpu.py
test/srt/test_gpt_oss_4gpu.py
+2
-8
No files found.
scripts/ci/ci_install_dependency.sh
View file @
de921733
...
@@ -47,8 +47,8 @@ $PIP_CMD install -e "python[dev]" --extra-index-url https://download.pytorch.org
...
@@ -47,8 +47,8 @@ $PIP_CMD install -e "python[dev]" --extra-index-url https://download.pytorch.org
if
[
"
$IS_BLACKWELL
"
=
"1"
]
;
then
if
[
"
$IS_BLACKWELL
"
=
"1"
]
;
then
# TODO auto determine sgl-kernel version
# TODO auto determine sgl-kernel version
SGL_KERNEL_VERSION
=
0.3.
2
SGL_KERNEL_VERSION
=
0.3.
8
$PIP_CMD
install
https://github.com/sgl-project/whl/releases/download/v
${
SGL_KERNEL_VERSION
}
/sgl_kernel-
${
SGL_KERNEL_VERSION
}
-cp3
9
-abi3-manylinux2014_x86_64
.whl
--force-reinstall
$PIP_INSTALL_SUFFIX
$PIP_CMD
install
https://github.com/sgl-project/whl/releases/download/v
${
SGL_KERNEL_VERSION
}
/sgl_kernel-
${
SGL_KERNEL_VERSION
}
+cu128
-cp3
10
-abi3-manylinux2014_x86_64.whl
--force-reinstall
$PIP_INSTALL_SUFFIX
fi
fi
# Show current packages
# Show current packages
...
...
test/srt/run_suite.py
View file @
de921733
...
@@ -139,6 +139,7 @@ suites = {
...
@@ -139,6 +139,7 @@ suites = {
],
],
"per-commit-8-gpu-b200"
:
[
"per-commit-8-gpu-b200"
:
[
# add more here
# add more here
TestFile
(
"test_gpt_oss_4gpu.py"
,
600
),
],
],
"per-commit-4-gpu-deepep"
:
[
"per-commit-4-gpu-deepep"
:
[
TestFile
(
"ep/test_deepep_small.py"
,
531
),
TestFile
(
"ep/test_deepep_small.py"
,
531
),
...
...
test/srt/test_gpt_oss_4gpu.py
View file @
de921733
...
@@ -9,10 +9,7 @@ class TestGptOss4Gpu(BaseTestGptOss):
...
@@ -9,10 +9,7 @@ class TestGptOss4Gpu(BaseTestGptOss):
model_variant
=
"120b"
,
model_variant
=
"120b"
,
quantization
=
"bf16"
,
quantization
=
"bf16"
,
expected_score_of_reasoning_effort
=
{
expected_score_of_reasoning_effort
=
{
"low"
:
0.61
,
"low"
:
0.60
,
# remove to speed up
# "medium": 0.61,
# "high": 0.61,
},
},
other_args
=
[
"--tp"
,
"4"
,
"--cuda-graph-max-bs"
,
"200"
],
other_args
=
[
"--tp"
,
"4"
,
"--cuda-graph-max-bs"
,
"200"
],
)
)
...
@@ -22,10 +19,7 @@ class TestGptOss4Gpu(BaseTestGptOss):
...
@@ -22,10 +19,7 @@ class TestGptOss4Gpu(BaseTestGptOss):
model_variant
=
"120b"
,
model_variant
=
"120b"
,
quantization
=
"mxfp4"
,
quantization
=
"mxfp4"
,
expected_score_of_reasoning_effort
=
{
expected_score_of_reasoning_effort
=
{
"low"
:
0.61
,
"low"
:
0.60
,
# remove to speed up
# "medium": 0.61,
# "high": 0.61,
},
},
other_args
=
[
other_args
=
[
"--tp"
,
"--tp"
,
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment