Skip to content

Commit 7b0213b

Browse files
committed
fix ci
Signed-off-by: wangli <wangli858794774@gmail.com>
1 parent 3275f77 commit 7b0213b

File tree

3 files changed

+6
-19
lines changed

3 files changed

+6
-19
lines changed

fusion_result.json

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1 @@
1+
null

tests/conftest.py

Lines changed: 1 addition & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -53,11 +53,7 @@ def cleanup_dist_env_and_memory():
5353
destroy_distributed_environment()
5454
gc.collect()
5555
torch.npu.empty_cache()
56-
try:
57-
torch._C._host_emptyCache()
58-
except AttributeError:
59-
logger.warning(
60-
"torch._C._host_emptyCache() only available in Pytorch >=2.5")
56+
6157

6258
class VllmRunner:
6359

tests/test_offline_inference.py

Lines changed: 4 additions & 14 deletions
Original file line numberDiff line numberDiff line change
@@ -38,13 +38,7 @@
3838
@pytest.mark.parametrize("use_v1", ["1", "0"])
3939
@pytest.mark.parametrize("dtype", ["half", "float16"])
4040
@pytest.mark.parametrize("max_tokens", [5])
41-
def test_models(
42-
model: str,
43-
use_v1: str,
44-
dtype: str,
45-
max_tokens: int,
46-
monkeypatch: pytest.MonkeyPatch
47-
) -> None:
41+
def test_models(model: str, use_v1: str, dtype: str, max_tokens: int, monkeypatch: pytest.MonkeyPatch) -> None:
4842
with monkeypatch.context() as m:
4943
m.setenv("VLLM_USE_V1", use_v1)
5044
if use_v1 == '1':
@@ -69,13 +63,9 @@ def test_models(
6963
@pytest.mark.parametrize("model, distributed_executor_backend", [
7064
("Qwen/QwQ-32B", "mp"),
7165
])
72-
def test_models_distributed(
73-
vllm_runner,
74-
use_v1: str,
75-
model: str,
76-
distributed_executor_backend: str,
77-
monkeypatch: pytest.MonkeyPatch
78-
) -> None:
66+
def test_models_distributed(vllm_runner, use_v1: str, model: str,
67+
distributed_executor_backend: str,
68+
monkeypatch: pytest.MonkeyPatch) -> None:
7969
with monkeypatch.context() as m:
8070
m.setenv("VLLM_USE_V1", use_v1)
8171
if use_v1 == '1':

0 commit comments

Comments
 (0)