|
37 | 37 |
|
38 | 38 | @pytest.mark.parametrize("model", MODELS) |
39 | 39 | @patch.dict(os.environ, {"HCCL_BUFFSIZE": "500"}) |
40 | | -def test_external_launcher(model): |
| 40 | +def test_qwen_external_launcher(model): |
41 | 41 | script = Path( |
42 | 42 | __file__ |
43 | 43 | ).parent.parent.parent.parent / "examples" / "offline_external_launcher.py" |
@@ -78,7 +78,7 @@ def test_external_launcher(model): |
78 | 78 |
|
79 | 79 |
|
80 | 80 | @pytest.mark.parametrize("model", MOE_MODELS) |
81 | | -def test_external_launcher_moe_ep_tp2(model): |
| 81 | +def test_qwen_moe_external_launcher_ep(model): |
82 | 82 | script = Path( |
83 | 83 | __file__ |
84 | 84 | ).parent.parent.parent.parent / "examples" / "offline_external_launcher.py" |
@@ -109,7 +109,7 @@ def test_external_launcher_moe_ep_tp2(model): |
109 | 109 |
|
110 | 110 |
|
111 | 111 | @patch.dict(os.environ, {"VLLM_ASCEND_ENABLE_NZ": "0"}) |
112 | | -def test_external_launcher_with_sleepmode_level1(): |
| 112 | +def test_qwen_external_launcher_with_sleepmode(): |
113 | 113 | script = Path( |
114 | 114 | __file__ |
115 | 115 | ).parent.parent.parent.parent / "examples" / "offline_external_launcher.py" |
@@ -154,7 +154,7 @@ def test_external_launcher_with_sleepmode_level1(): |
154 | 154 |
|
155 | 155 |
|
156 | 156 | @patch.dict(os.environ, {"VLLM_ASCEND_ENABLE_NZ": "0"}) |
157 | | -def test_external_launcher_with_sleepmode_level2(): |
| 157 | +def test_qwen_external_launcher_with_sleepmode_level2(): |
158 | 158 | script = Path( |
159 | 159 | __file__ |
160 | 160 | ).parent.parent.parent.parent / "examples" / "offline_external_launcher.py" |
@@ -210,7 +210,7 @@ def test_external_launcher_with_sleepmode_level2(): |
210 | 210 | "VLLM_ASCEND_ENABLE_MATMUL_ALLREDUCE": "1", |
211 | 211 | "HCCL_BUFFSIZE": "500" |
212 | 212 | }) |
213 | | -def test_external_launcher_with_matmul_allreduce(model): |
| 213 | +def test_qwen_external_launcher_with_matmul_allreduce(model): |
214 | 214 | script = Path( |
215 | 215 | __file__ |
216 | 216 | ).parent.parent.parent.parent / "examples" / "offline_external_launcher.py" |
|
0 commit comments