Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in
Toggle navigation
Menu
Open sidebar
OpenDAS
vllm_cscc
Commits
5b4b42c0
Unverified
Commit
5b4b42c0
authored
Dec 04, 2025
by
Doug Smith
Committed by
GitHub
Dec 04, 2025
Browse files
Mark DBO test as flaky on b200 for Distributed B200 test (#29913)
Signed-off-by:
dougbtv
<
dosmith@redhat.com
>
parent
cc050558
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
19 additions
and
0 deletions
+19
-0
tests/v1/distributed/test_dbo.py
tests/v1/distributed/test_dbo.py
+19
-0
No files found.
tests/v1/distributed/test_dbo.py
View file @
5b4b42c0
...
@@ -9,10 +9,22 @@ correctly with the DeepSeek-V2-Lite model using GSM8K evaluation.
...
@@ -9,10 +9,22 @@ correctly with the DeepSeek-V2-Lite model using GSM8K evaluation.
"""
"""
import
pytest
import
pytest
import
torch
from
tests.evals.gsm8k.gsm8k_eval
import
evaluate_gsm8k
from
tests.evals.gsm8k.gsm8k_eval
import
evaluate_gsm8k
from
tests.utils
import
RemoteOpenAIServer
from
tests.utils
import
RemoteOpenAIServer
# Detect Blackwell / B200 (compute capability 10.x)
try
:
if
torch
.
cuda
.
is_available
():
cap
=
torch
.
cuda
.
get_device_capability
(
0
)
IS_BLACKWELL
=
cap
[
0
]
>=
10
else
:
IS_BLACKWELL
=
False
except
Exception
:
# Be conservative: if we can't detect, don't xfail by default
IS_BLACKWELL
=
False
MODEL_NAME
=
"deepseek-ai/DeepSeek-V2-Lite-Chat"
MODEL_NAME
=
"deepseek-ai/DeepSeek-V2-Lite-Chat"
DP_SIZE
=
2
DP_SIZE
=
2
...
@@ -33,6 +45,13 @@ DEEPEP_BACKENDS = [
...
@@ -33,6 +45,13 @@ DEEPEP_BACKENDS = [
@
pytest
.
mark
.
parametrize
(
"all2all_backend"
,
DEEPEP_BACKENDS
)
@
pytest
.
mark
.
parametrize
(
"all2all_backend"
,
DEEPEP_BACKENDS
)
@
pytest
.
mark
.
xfail
(
IS_BLACKWELL
,
reason
=
(
"Temporary: DBO accuracy unstable on Blackwell "
"(doesn't meet expectation of MIN_ACCURACY = 0.62)"
),
)
def
test_dbo_dp_ep_gsm8k
(
all2all_backend
:
str
,
num_gpus_available
):
def
test_dbo_dp_ep_gsm8k
(
all2all_backend
:
str
,
num_gpus_available
):
"""
"""
Test DBO with DP+EP using GSM8K evaluation.
Test DBO with DP+EP using GSM8K evaluation.
...
...
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