Skip to content

Commit 26892df

Browse files
committed
fix pass manager test
Signed-off-by: Luka Govedič <lgovedic@redhat.com>
1 parent 0d6e550 commit 26892df

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

tests/compile/test_pass_manager.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -7,7 +7,7 @@
77

88
from vllm.compilation.inductor_pass import CallableInductorPass, InductorPass
99
from vllm.compilation.pass_manager import PostGradPassManager
10-
from vllm.config import VllmConfig
10+
from vllm.config import ModelConfig, VllmConfig
1111

1212

1313
# dummy custom pass that doesn't inherit
@@ -42,7 +42,8 @@ def __call__(self, graph: torch.fx.graph.Graph) -> None:
4242
],
4343
)
4444
def test_pass_manager_uuid(callable):
45-
config = VllmConfig()
45+
# Some passes need dtype to be set
46+
config = VllmConfig(model_config=ModelConfig(dtype=torch.bfloat16))
4647

4748
pass_manager = PostGradPassManager()
4849
pass_manager.configure(config)

0 commit comments

Comments
 (0)