Skip to content

Commit 1b68223

Browse files
[pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
1 parent 867acf8 commit 1b68223

File tree

1 file changed

+4
-2
lines changed
  • neural_compressor/torch/algorithms/mx_quant

1 file changed

+4
-2
lines changed

neural_compressor/torch/algorithms/mx_quant/mx.py

+4-2
Original file line numberDiff line numberDiff line change
@@ -19,8 +19,10 @@
1919
# limitations under the License.
2020

2121
import torch
22-
from neural_compressor.torch.utils import set_module, logger
2322
from torch.nn import functional as F
23+
24+
from neural_compressor.torch.utils import logger, set_module
25+
2426
from .utils import quantize_elemwise_op, quantize_mx_op
2527

2628

@@ -123,7 +125,7 @@ def mx_quantize(
123125
logger.debug(f"MX quantized module:{name, m}")
124126
log_msg = (
125127
f"MX quantization config: w_dtype={config[(name, type(m).__name__)].w_dtype}, "
126-
+ f"config[(name, type(m).__name__)].act_dtype, "
128+
+ "config[(name, type(m).__name__)].act_dtype, "
127129
+ f"out_dtype={config[(name, type(m).__name__)].out_dtype}"
128130
)
129131
logger.debug(log_msg)

0 commit comments

Comments
 (0)