biondizzle/vllm
1
0
Fork 0
Code Issues Pull Requests Actions 2 Packages Projects Releases Wiki Activity

Remove last level references not removed in #26355 (#27260)

Browse Source 
Signed-off-by: Harry Mellor <19981378+hmellor@users.noreply.github.com>
This commit is contained in:
Harry Mellor
2025-10-22 10:18:17 +01:00
committed by GitHub
parent ed540d6d4c
commit 8f18feb191
6 changed files with 13 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
tests/model_executor/test_enabled_custom_ops.py
View File

@@ -36,7 +36,7 @@ class Relu3(ReLUSquaredActivation):
@pytest.mark.parametrize(
"env, torch_level, backend, ops_enabled, default_on",
"env, compilation_mode, backend, ops_enabled, default_on",
[
# Default values based on compile level
# - All by default (no Inductor compilation)
@@ -77,7 +77,7 @@ class Relu3(ReLUSquaredActivation):
)
def test_enabled_ops(
env: str | None,
torch_level: int,
compilation_mode: int,
backend: str,
ops_enabled: list[int],
default_on: bool,
@@ -85,7 +85,7 @@ def test_enabled_ops(
custom_ops = env.split(",") if env else []
vllm_config = VllmConfig(
compilation_config=CompilationConfig(
backend=backend, level=torch_level, custom_ops=custom_ops
backend=backend, mode=compilation_mode, custom_ops=custom_ops
)
)
with set_current_vllm_config(vllm_config):