Skip to content

Commit b749a91

Browse files
committed
fix test
Signed-off-by: cascade812 <[email protected]>
1 parent c05a9a4 commit b749a91

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

tests/compile/test_fusion.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -125,5 +125,5 @@ def test_fusion_rmsnorm_quant(dtype, hidden_size, num_tokens, eps, static,
125125
find_auto_fn_maybe)
126126

127127
# In post-nodes, fused kernels should be there and fp8 quant should not
128-
backend.check_before_ops(model.ops_in_model_after(), find_auto_fn,
129-
find_auto_fn_maybe)
128+
backend.check_after_ops(model.ops_in_model_after(), find_auto_fn,
129+
find_auto_fn_maybe)

0 commit comments

Comments
 (0)