Skip to content

Commit 35f3da2

Browse files
committed
update comment
Signed-off-by: Jingyu Xin <[email protected]>
1 parent 94ec97d commit 35f3da2

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

modelopt/torch/quantization/nn/modules/tensor_quantizer.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -960,6 +960,7 @@ def forward(self, inputs):
960960
and self._fake_quant
961961
):
962962
# Reshape is required if the logic is not handled in the simulation kernel
963+
# Only MX format and NVFP4 reshape are currently supported by the kernel.
963964
self._setup_for_blockquant(inputs)
964965
setattr(self, "_original_input_shape", inputs.shape)
965966
inputs = self._process_for_blockquant(inputs)

0 commit comments

Comments
 (0)