fix sgl-kernel unit tests (#5666)

2025-04-23 01:18:30 -07:00
parent e62c49557d
commit 15fabcc07f
9 changed files with 313 additions and 0 deletions
--- a/sgl-kernel/tests/test_moe_fused_gate.py
+++ b/sgl-kernel/tests/test_moe_fused_gate.py
@@ -48,6 +48,7 @@ def test_moe_fused_gate_combined(seq_length, dtype, params, n_share_experts_fusi
        topk_group=topk_group,
        compiled=False,
        n_share_experts_fusion=n_share_experts_fusion,
+        routed_scaling_factor=2.5,
    )

    # When n_share_experts_fusion > 0, ignore the comparison of the last topk dimension