We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 3f3a23c commit 89ccf4cCopy full SHA for 89ccf4c
lightllm/common/fused_moe/grouped_fused_moe.py
@@ -467,6 +467,8 @@ def grouped_matmul_kernel(
467
offs_am = tile_m_idx * BLOCK_SIZE_M + tl.arange(0, BLOCK_SIZE_M)
468
token_mask = offs_am < cur_m
469
470
+ assert (OUT_SORTED and TOKEN_INPUT_USE_TMA) is False
471
+
472
if not OUT_SORTED or not TOKEN_INPUT_USE_TMA:
473
a_m_index = tl.load(
474
expert_to_token_index + expert_id * expert_to_token_index_stride_0 + offs_am,
0 commit comments