From eedb2a2a102b47d527071b9acfb9edd541ef5de6 Mon Sep 17 00:00:00 2001 From: elvischenv <219235043+elvischenv@users.noreply.github.com> Date: Sat, 6 Sep 2025 04:13:42 +0800 Subject: [PATCH] [Bugfix] Fix silu_mul+quant fusion test (#24341) Signed-off-by: elvischenv <219235043+elvischenv@users.noreply.github.com> --- tests/compile/test_silu_mul_quant_fusion.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/tests/compile/test_silu_mul_quant_fusion.py b/tests/compile/test_silu_mul_quant_fusion.py index e16d1725e6ad..731ceeb905f6 100644 --- a/tests/compile/test_silu_mul_quant_fusion.py +++ b/tests/compile/test_silu_mul_quant_fusion.py @@ -118,7 +118,8 @@ def test_fusion_silu_and_mul_quant(num_tokens, hidden_size, model_class, fusion_pass = ActivationQuantFusionPass(config) backend = TestBackend(NoOpEliminationPass(config), fusion_pass) - model = model_class(hidden_size, cuda_force_torch) + model = model_class(hidden_size=hidden_size, + cuda_force_torch=cuda_force_torch) # First dimension dynamic x = torch.rand(num_tokens, hidden_size * 2)