diff --git a/python/sglang/kernels/ops/moe/ep_moe_kernels.py b/python/sglang/kernels/ops/moe/ep_moe_kernels.py index 938b82951..9268572e3 100644 --- a/python/sglang/kernels/ops/moe/ep_moe_kernels.py +++ b/python/sglang/kernels/ops/moe/ep_moe_kernels.py @@ -241,7 +241,7 @@ def pre_reorder_triton_kernel_for_cutlass_moe( for idx in range(topk): expert_id = tl.load(token_topk_ids_ptr + idx) if expert_id != num_local_experts: - dst_idx = tl.load(token_src2dst_ptr + idx) + dst_idx = tl.load(token_src2dst_ptr + idx).to(tl.int64) tl.store(dst_ptr_offs + dst_idx * hidden_size, out_data, mask=mask)