Skip to content

Commit 684175d

Browse files
committed
changed callable_decorated to callable_uuid
Signed-off-by: JovanSardinha <jovan.sardinha@gmail.com>
1 parent 6d053f0 commit 684175d

File tree

1 file changed

+3
-4
lines changed

1 file changed

+3
-4
lines changed

tests/compile/test_pass_manager.py

Lines changed: 3 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -4,7 +4,6 @@
44

55
import pytest
66
import torch
7-
from torch._inductor.codecache import BypassFxGraphCache
87

98
from vllm.compilation.inductor_pass import CallableInductorPass, InductorPass
109
from 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
)

0 commit comments

Comments
 (0)