Skip to content

Commit 1e0806f

Browse files
authored
Fix DeepGEMM masked cannot be run on groups not being multiple or 4 (#5340)
1 parent 2c11f9c commit 1e0806f

File tree

1 file changed

+0
-3
lines changed
  • python/sglang/srt/layers/moe/ep_moe

1 file changed

+0
-3
lines changed

python/sglang/srt/layers/moe/ep_moe/layer.py

Lines changed: 0 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -972,9 +972,6 @@ def forward_deepgemm_masked(
972972
):
973973
assert self.quant_method is not None
974974
assert self.activation == "silu"
975-
assert (
976-
hidden_states_fp8[0].size(0) % 4 == 0
977-
), f"TMA alignment error: {hidden_states_fp8[0].size(0)}"
978975

979976
# GroupGemm-0
980977
num_groups, m, k = hidden_states_fp8[0].size()

0 commit comments

Comments
 (0)