Skip to content

Commit 07e1b7d

Browse files
Update test_offline_inference_distributed.py
1 parent 8973189 commit 07e1b7d

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

tests/multicard/test_offline_inference_distributed.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -84,7 +84,10 @@ def test_models_distributed_topk() -> None:
8484
) as vllm_model:
8585
vllm_model.generate(example_prompts, sampling_params)
8686

87-
87+
@pytest.mark.skip(
88+
reason=
89+
"deepseek dbo dose not consider the support on half precision float, will enable this ut after we actually support it"
90+
)
8891
@patch.dict(os.environ, {"VLLM_ASCEND_ENABLE_DBO": "1"})
8992
def test_models_distributed_DeepSeek_dbo():
9093
example_prompts = ["The president of the United States is"] * 41

0 commit comments

Comments
 (0)