We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent c22a283 commit 219ecd8Copy full SHA for 219ecd8
tensorrt_llm/_torch/modules/fused_moe/quantization.py
@@ -6,8 +6,8 @@
6
import torch.nn.functional as F
7
from torch import nn
8
9
-from tensorrt_llm import logger
10
from tensorrt_llm._utils import get_sm_version
+from tensorrt_llm.logger import logger
11
from tensorrt_llm.quantization.utils.fp4_utils import (
12
float4_sf_dtype, get_reorder_rows_for_gated_act_gemm_row_indices,
13
get_shuffle_matrix_a_row_indices, get_shuffle_matrix_sf_a_row_indices)
0 commit comments