diff --git a/tests/compile/piecewise/test_multiple_graphs.py b/tests/compile/piecewise/test_multiple_graphs.py index 967167e2d160d..f6256018348c5 100644 --- a/tests/compile/piecewise/test_multiple_graphs.py +++ b/tests/compile/piecewise/test_multiple_graphs.py @@ -17,7 +17,7 @@ from vllm.envs import VLLM_USE_V1 from vllm.forward_context import set_forward_context # This import automatically registers torch ops for testing (like silly.attention) -import tests.compile.test_operations +import tests.compile.testing_ops BATCH_SIZE = 32 MLP_SIZE = 128 diff --git a/tests/compile/piecewise/test_simple.py b/tests/compile/piecewise/test_simple.py index a1535ff6c5ebc..4df709b1ba863 100644 --- a/tests/compile/piecewise/test_simple.py +++ b/tests/compile/piecewise/test_simple.py @@ -16,7 +16,7 @@ from vllm.envs import VLLM_USE_V1 from vllm.forward_context import BatchDescriptor, set_forward_context # This import also automatically registers torch ops for testing (like silly.attention) -from tests.compile.test_operations import ( +from tests.compile.testing_ops import ( get_global_counter, reset_global_counter ) diff --git a/tests/compile/piecewise/test_toy_llama.py b/tests/compile/piecewise/test_toy_llama.py index 6a2947fb2b715..5088322ffab27 100644 --- a/tests/compile/piecewise/test_toy_llama.py +++ b/tests/compile/piecewise/test_toy_llama.py @@ -22,7 +22,7 @@ from vllm.config import (CompilationConfig, CompilationLevel, CUDAGraphMode, from vllm.forward_context import BatchDescriptor, set_forward_context # This import automatically registers torch ops for testing (like silly.attention) -import tests.compile.test_operations +import tests.compile.testing_ops @dataclass diff --git a/tests/compile/test_operations.py b/tests/compile/testing_ops.py similarity index 100% rename from tests/compile/test_operations.py rename to tests/compile/testing_ops.py