minor test fix

Signed-off-by: Sage Moore <[email protected]>
vllm-project · Dec 20, 2024 · 4a0ac7e · 4a0ac7e
1 parent 008b725
commit 4a0ac7e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/compile/test_functionalization.py b/tests/compile/test_functionalization.py
@@ -86,7 +86,7 @@ def test_fix_functionalization(model: str, quant_key: QuantKey,
     # and replaced by fused quantized ops in RMS_QUANT_OPS.
     rms_ops = [FUSED_OPS[(quant_key, True)], FUSED_OPS[(quant_key, False)]
                ] if do_fusion else [RMS_OP]
-    silu_mul_ops = SILU_MUL_QUANT_OP if do_fusion and \
+    silu_mul_ops = [SILU_MUL_QUANT_OP] if do_fusion and \
         quant_key == kFp8StaticTensorSym else [
         SILU_MUL_OP
     ]