Skip to content

Commit 68375ed

Browse files
committed
Fix typo.
Signed-off-by: Bo Li <22713281+bobboli@users.noreply.github.com>
1 parent f5b86ad commit 68375ed

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tensorrt_llm/_torch/modules/fused_moe/fused_moe_cutlass.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -532,7 +532,7 @@ def forward_chunk(
532532

533533
# Optionally provide an output tensor to fused_moe so it writes directly to our buffer
534534
moe_output: Optional[torch.Tensor] = None
535-
if self.enable_alltoall and self.moe_alltoall_backend == "NVLINK_ONE_SIDEDz":
535+
if self.enable_alltoall and self.moe_alltoall_backend == "NVLINK_ONE_SIDED":
536536
# Retrieve a workspace-backed output tensor sized by runtime tokens
537537
runtime_max_tokens_per_rank = max(
538538
all_rank_num_tokens) if all_rank_num_tokens else x.shape[0]

0 commit comments

Comments
 (0)