Skip to content

Commit eedb2a2

Browse files
authored
[Bugfix] Fix silu_mul+quant fusion test (#24341)
Signed-off-by: elvischenv <219235043+elvischenv@users.noreply.github.com>
1 parent 23a6c52 commit eedb2a2

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

tests/compile/test_silu_mul_quant_fusion.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -118,7 +118,8 @@ def test_fusion_silu_and_mul_quant(num_tokens, hidden_size, model_class,
118118
fusion_pass = ActivationQuantFusionPass(config)
119119

120120
backend = TestBackend(NoOpEliminationPass(config), fusion_pass)
121-
model = model_class(hidden_size, cuda_force_torch)
121+
model = model_class(hidden_size=hidden_size,
122+
cuda_force_torch=cuda_force_torch)
122123

123124
# First dimension dynamic
124125
x = torch.rand(num_tokens, hidden_size * 2)

0 commit comments

Comments
 (0)