File tree Expand file tree Collapse file tree 1 file changed +3
-4
lines changed Expand file tree Collapse file tree 1 file changed +3
-4
lines changed Original file line number Diff line number Diff line change 44
55import pytest
66import torch
7- from torch ._inductor .codecache import BypassFxGraphCache
87
98from vllm .compilation .inductor_pass import CallableInductorPass , InductorPass
109from vllm .compilation .pass_manager import PostGradPassManager
@@ -15,15 +14,15 @@ def simple_callable(graph: torch.fx.Graph):
1514 pass
1615
1716
18- callable_decorated = CallableInductorPass (simple_callable ,
19- InductorPass .hash_source (__file__ ))
17+ callable_uuid = CallableInductorPass (simple_callable ,
18+ InductorPass .hash_source (__file__ ))
2019
2120
2221@pytest .mark .parametrize (
2322 "works, callable" ,
2423 [
2524 (False , simple_callable ),
26- (True , callable_decorated ),
25+ (True , callable_uuid ),
2726 (True , CallableInductorPass (simple_callable )),
2827 ],
2928)
You can’t perform that action at this time.
0 commit comments