Skip to content

Commit e7ac015

Browse files
committed
fix typo
Signed-off-by: Siyuan Fu <[email protected]>
1 parent 7995977 commit e7ac015

File tree

1 file changed

+2
-3
lines changed

1 file changed

+2
-3
lines changed

csrc/trtllm_fused_moe_routing_llama4.cu

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -259,11 +259,10 @@ __global__ void __launch_bounds__(WarpSize) routingIndicesWarpKernel(KernelParam
259259
// of registers
260260
auto localExpertExtent = params.mNumLocalExperts << params.mLocalExpertsStrideLog2;
261261
int32_t finalExpertOffset[ExpertsPerThread];
262-
int32_t finalExpertOffset0;
263262
if constexpr (KernelParams::isPow2) {
264-
finalExpertOffset0 = mulLog2<int32_t>(ctaOffset, params.mPaddingLog2);
263+
finalExpertOffset[0] = mulLog2<int32_t>(ctaOffset, params.mPaddingLog2);
265264
} else {
266-
finalExpertOffset0 = mulTileN<int32_t>(ctaOffset, params.mTileTokensDim);
265+
finalExpertOffset[0] = mulTileN<int32_t>(ctaOffset, params.mTileTokensDim);
267266
}
268267
#pragma unroll
269268
for (int ii = 1; ii < ExpertsPerThread; ++ii) {

0 commit comments

Comments
 (0)