Skip to content

Commit 17155b1

Browse files
committed
Fix typo
Signed-off-by: Dongfeng Yu <[email protected]>
1 parent 7794e13 commit 17155b1

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tensorrt_llm/_torch/modules/fused_moe/fused_moe_triton.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -643,7 +643,7 @@ def swizzle_weight_and_scale(w: torch.Tensor, w_scale: torch.Tensor):
643643
mx_axis=1)
644644
scale_layout, scale_layout_opts = layout.make_default_matmul_mxfp4_w_scale_layout(
645645
mx_axis=1, num_warps=num_warps)
646-
# Swizzeling path is broken for H20
646+
# swizzling path is broken for H20
647647
if torch.cuda.get_device_name() == "NVIDIA H20":
648648
from triton_kernels.tensor_details.layout_details.strided import \
649649
StridedLayout

0 commit comments

Comments
 (0)