Skip to content

Commit 7584bd3

Browse files
committed
enable deepseek CI on v1
Signed-off-by: MengqingCao <cmq0113@163.com>
1 parent 3238e83 commit 7584bd3

File tree

2 files changed

+8
-9
lines changed

2 files changed

+8
-9
lines changed

.github/workflows/vllm_ascend_test.yaml

Lines changed: 2 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -116,8 +116,7 @@ jobs:
116116
pytest -sv tests/ops
117117
pytest -sv tests/compile
118118
else
119-
pytest -sv -k "QwQ" tests/multicard/test_offline_inference_distributed.py
120-
pytest -sv tests/multicard/test_ilama_lora_tp2.py
119+
pytest -sv tests/multicard/
121120
pytest -sv tests/ops
122121
pytest -sv tests/compile
123122
fi
@@ -131,9 +130,7 @@ jobs:
131130
pytest -sv tests/singlecard/test_offline_inference.py
132131
pytest -sv tests/ops
133132
else
134-
pytest -sv tests/multicard/test_ilama_lora_tp2.py
135-
pytest -sv -k "QwQ" tests/multicard/test_offline_inference_distributed.py
136-
pytest -sv -k "DeepSeek" tests/multicard/test_offline_inference_distributed.py
133+
pytest -sv tests/multicard/
137134
pytest -sv tests/ops
138135
fi
139136

tests/multicard/test_offline_inference_distributed.py

Lines changed: 6 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -30,11 +30,13 @@
3030
os.environ["PYTORCH_NPU_ALLOC_CONF"] = "max_split_size_mb:256"
3131
os.environ["VLLM_USE_MODELSCOPE"] = "True"
3232

33+
MODELS = ["Qwen/QwQ-32B", "deepseek-ai/DeepSeek-V2-Lite"]
34+
DIST_EXECUTOR_BACKENDS = ["mp", "ray"]
3335

34-
@pytest.mark.parametrize("model, distributed_executor_backend", [
35-
("Qwen/QwQ-32B", "mp"),
36-
("deepseek-ai/DeepSeek-V2-Lite", "mp"),
37-
])
36+
37+
@pytest.mark.parametrize("model", MODELS)
38+
@pytest.mark.parametrize("distributed_executor_backend",
39+
DIST_EXECUTOR_BACKENDS)
3840
def test_models_distributed(model: str,
3941
distributed_executor_backend: str) -> None:
4042
example_prompts = [

0 commit comments

Comments
 (0)