Skip to content

Commit 89ccf4c

Browse files
author
wangzaijun
committed
fix
1 parent 3f3a23c commit 89ccf4c

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

lightllm/common/fused_moe/grouped_fused_moe.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -467,6 +467,8 @@ def grouped_matmul_kernel(
467467
offs_am = tile_m_idx * BLOCK_SIZE_M + tl.arange(0, BLOCK_SIZE_M)
468468
token_mask = offs_am < cur_m
469469

470+
assert (OUT_SORTED and TOKEN_INPUT_USE_TMA) is False
471+
470472
if not OUT_SORTED or not TOKEN_INPUT_USE_TMA:
471473
a_m_index = tl.load(
472474
expert_to_token_index + expert_id * expert_to_token_index_stride_0 + offs_am,

0 commit comments

Comments
 (0)