diff --git a/python/sglang/srt/layers/moe/ep_moe/layer.py b/python/sglang/srt/layers/moe/ep_moe/layer.py index a35d0b8d0..a5a9eb738 100644 --- a/python/sglang/srt/layers/moe/ep_moe/layer.py +++ b/python/sglang/srt/layers/moe/ep_moe/layer.py @@ -972,9 +972,6 @@ class DeepEPMoE(EPMoE): ): assert self.quant_method is not None assert self.activation == "silu" - assert ( - hidden_states_fp8[0].size(0) % 4 == 0 - ), f"TMA alignment error: {hidden_states_fp8[0].size(0)}" # GroupGemm-0 num_groups, m, k = hidden_states_fp8[0].size()