Rename test_operations.py to testing_ops.py and update all imports
Co-authored-by: ProExpertProg <11367180+ProExpertProg@users.noreply.github.com>
This commit is contained in:
@ -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
|
||||
|
||||
@ -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
|
||||
)
|
||||
|
||||
|
||||
@ -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
|
||||
|
||||
Reference in New Issue
Block a user