Skip to content

Commit

Permalink
fix config setting
Browse files Browse the repository at this point in the history
Signed-off-by: Mengni Wang <[email protected]>
  • Loading branch information
mengniwang95 committed Aug 26, 2024
1 parent d59fdca commit 27eae66
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions onnx_neural_compressor/quantization/config.py
Original file line number Diff line number Diff line change
Expand Up @@ -802,10 +802,10 @@ def to_config_mapping(self, config_list: List[BaseConfig] = None, model_info: li
global_config = config.get_params_dict()
op_type_config_dict, op_name_config_dict = config._get_op_name_op_type_config()
for op_name, op_type in model_info:
if op_name in self.nodes_to_exclude:
continue
if op_type not in self.white_list:
continue
if any([re.match(exclude_name, op_name) for exclude_name in self.nodes_to_exclude]):
continue
if op_type == "MatMul":
last_matmul = op_name
if global_config is not None:
Expand Down

0 comments on commit 27eae66

Please sign in to comment.