We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
op_name
1 parent ba47985 commit 911ccd3Copy full SHA for 911ccd3
neural_compressor/common/base_config.py
@@ -412,7 +412,9 @@ def to_config_mapping(
412
if op_type in op_type_config_dict:
413
config_mapping[(op_name, op_type)] = op_name_config_dict[op_type]
414
for op_name_pattern in op_name_config_dict:
415
- if re.match(op_name_pattern, op_name):
+ 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
418
config_mapping[(op_name, op_type)] = op_name_config_dict[op_name_pattern]
419
return config_mapping
420
0 commit comments