Skip to content

Commit 219ecd8

Browse files
committed
fixing a typo introduced in PR NVIDIA#6616
Signed-off-by: Rakib Hasan <[email protected]>
1 parent c22a283 commit 219ecd8

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tensorrt_llm/_torch/modules/fused_moe/quantization.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -6,8 +6,8 @@
66
import torch.nn.functional as F
77
from torch import nn
88

9-
from tensorrt_llm import logger
109
from tensorrt_llm._utils import get_sm_version
10+
from tensorrt_llm.logger import logger
1111
from tensorrt_llm.quantization.utils.fp4_utils import (
1212
float4_sf_dtype, get_reorder_rows_for_gated_act_gemm_row_indices,
1313
get_shuffle_matrix_a_row_indices, get_shuffle_matrix_sf_a_row_indices)

0 commit comments

Comments
 (0)