Skip to content

Commit abaa038

Browse files
committed
[SW-192358] Remove HQT reference in INC
Change-Id: Ic25f9323486596fa2dc6d909cd568a37ab84dd5e
1 parent 56c03d8 commit abaa038

File tree

3 files changed

+4
-6
lines changed

3 files changed

+4
-6
lines changed

neural_compressor/torch/algorithms/fp8_quant/_core/utils.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -13,10 +13,10 @@ def update_mod_dict(config):
1313

1414
def print_init_info(config):
1515
import importlib.metadata
16-
versionStr = importlib.metadata.version('habana_quantization_toolkit')
16+
versionStr = importlib.metadata.version('neural_compressor_3x_pt')
1717
locationStr = versionStr.find('git') + 3
18-
logger.info("HQT Git revision = %s", versionStr[locationStr:])
19-
logger.info("HQT Configuration = %s", config)
18+
logger.info("neural_compressor_3x_pt Git revision = %s", versionStr[locationStr:])
19+
logger.info("neural_compressor_3x_pt Configuration = %s", config)
2020

2121
def is_substr(substr_list, target):
2222
return any([x in target for x in substr_list])

neural_compressor/torch/algorithms/fp8_quant/common.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -70,7 +70,7 @@ def create_mod_info_recursion(parent):
7070

7171

7272
def get_patched_mod_list():
73-
from habana_quantization_toolkit._core.common import mod_default_dict
73+
from ._core.common import mod_default_dict
7474

7575
patched_mod_list = []
7676
for patched_mod in mod_default_dict.values():

neural_compressor/torch/algorithms/fp8_quant/fp8_quant.py

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -54,8 +54,6 @@ def _convert(model, config_path):
5454

5555

5656
def _prepare(model, config_path):
57-
import habana_quantization_toolkit as hqt
58-
5957
# update mode to MEASURE
6058
config_path = update_mode(config_path, measure_step=True)
6159

0 commit comments

Comments
 (0)