diff --git a/tests/compile/test_functionalization.py b/tests/compile/test_functionalization.py index 871e388ec2212..1c8fe53baa0fb 100644 --- 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 ]