Skip to content

Commit a565152

Browse files
gmagogsfmRunkaiTao
authored andcommitted
[Bugfix]Fix a conditional to not check zero value (vllm-project#28754)
Signed-off-by: Yanan Cao <gmagogsfm@gmail.com> Signed-off-by: Runkai Tao <rt572@physics.rutgers.edu>
1 parent 7539f95 commit a565152

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

vllm/compilation/caching.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -116,7 +116,8 @@ def optimized_call(*example_inputs):
116116
the AOT compiled path.
117117
"""
118118
compile_inputs = [
119-
inp or example_inputs[i] for i, inp in enumerate(fn.example_inputs)
119+
inp if inp is not None else example_inputs[i]
120+
for i, inp in enumerate(fn.example_inputs)
120121
]
121122
with tracing(TracingContext(fake_mode)):
122123
fn.optimized_call = vllm_backend(

0 commit comments

Comments
 (0)