Skip to content

Commit 911ccd3

Browse files
authored
Add non-str op_name match workaround for IPEX (#1715)
Signed-off-by: Kaihui-intel <kaihui.tang@intel.com>
1 parent ba47985 commit 911ccd3

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

neural_compressor/common/base_config.py

+3-1
Original file line numberDiff line numberDiff line change
@@ -412,7 +412,9 @@ def to_config_mapping(
412412
if op_type in op_type_config_dict:
413413
config_mapping[(op_name, op_type)] = op_name_config_dict[op_type]
414414
for op_name_pattern in op_name_config_dict:
415-
if re.match(op_name_pattern, op_name):
415+
if isinstance(op_name, str) and re.match(op_name_pattern, op_name):
416+
config_mapping[(op_name, op_type)] = op_name_config_dict[op_name_pattern]
417+
elif op_name_pattern == op_name: # TODO: map ipex opname to stock pt op_name
416418
config_mapping[(op_name, op_type)] = op_name_config_dict[op_name_pattern]
417419
return config_mapping
418420

0 commit comments

Comments
 (0)