Skip to content

Commit b23bf4c

Browse files
hiworldwzjwangzaijun
andauthored
fix unit test for silu_and_mul kernel. (#1075)
Co-authored-by: wangzaijun <[email protected]>
1 parent 971b876 commit b23bf4c

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

unit_tests/common/fused_moe/test_moe_silu_and_mul_mix_quant_ep.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -45,6 +45,9 @@ def test_silu_and_mul_masked(expert_num, token_num, hidden_dim):
4545

4646
silu_and_mul_masked_post_quant_fwd(in_tensor, out_tensor, out_scale_tensor, quant_group_size, masked_m)
4747

48+
true_out_tensor = true_out_tensor.view(out_tensor.shape)
49+
true_out_scale_tensor = true_out_scale_tensor.view(out_scale_tensor.shape)
50+
4851
for expert_id, expert_token_num in enumerate(masked_m.cpu().numpy()):
4952
assert torch.allclose(
5053
true_out_tensor[expert_id, :expert_token_num, :].to(torch.float32),

0 commit comments

Comments
 (0)