Skip to content

Commit efb3324

Browse files
angelayialhridoy
authored andcommitted
[compile] Fix inductor partition config (vllm-project#26645)
Signed-off-by: angelayi <yiangela7@gmail.com>
1 parent 033977e commit efb3324

File tree

1 file changed

+1
-3
lines changed

1 file changed

+1
-3
lines changed

vllm/config/compilation.py

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -709,9 +709,7 @@ def is_attention_compiled_piecewise(self) -> bool:
709709
return self.level == CompilationLevel.PIECEWISE
710710

711711
# Inductor partition case
712-
return (
713-
self.level > CompilationLevel.NO_COMPILATION and self.backend == "inductor"
714-
)
712+
return self.level > CompilationLevel.NO_COMPILATION and self.use_inductor
715713

716714
def custom_op_log_check(self):
717715
"""

0 commit comments

Comments
 (0)