Comments

openvinotoolkit · Sep 25, 2024 · b4135c8 · b4135c8
1 parent 560ae29
commit b4135c8
Show file tree

Hide file tree

Showing 4 changed files with 4 additions and 8 deletions.
diff --git a/nncf/quantization/algorithms/min_max/onnx_backend.py b/nncf/quantization/algorithms/min_max/onnx_backend.py
@@ -269,5 +269,4 @@ def should_quantize_weight(weight_name: str, quantized_weight_names: Set[str]) -
         return weight_name not in quantized_weight_names
 
     def is_matmul_with_constant(self, node: NNCFNode, nncf_graph: NNCFGraph) -> bool:
-        assert node.metatype in self.mat_mul_metatypes
-        return node.layer_attributes.has_weight()
+        return node.metatype in self.mat_mul_metatypes and node.layer_attributes.has_weight()
diff --git a/nncf/quantization/algorithms/min_max/openvino_backend.py b/nncf/quantization/algorithms/min_max/openvino_backend.py
@@ -264,8 +264,7 @@ def get_weight_nodes(self, nncf_graph: NNCFGraph) -> List[NNCFNode]:
         ]
 
     def is_matmul_with_constant(self, node: NNCFNode, nncf_graph: NNCFGraph) -> bool:
-        assert node.metatype in self.mat_mul_metatypes
-        return node.layer_attributes is not None
+        return node.metatype in self.mat_mul_metatypes and node.layer_attributes is not None
 
     @staticmethod
     def get_weight_name(nncf_graph: NNCFGraph, target_point: OVTargetPoint) -> str:

diff --git a/nncf/quantization/algorithms/min_max/torch_backend.py b/nncf/quantization/algorithms/min_max/torch_backend.py
@@ -384,5 +384,4 @@ def get_weight_nodes(self, nncf_graph: NNCFGraph) -> List[NNCFNode]:
         return weight_nodes
 
     def is_matmul_with_constant(self, node: NNCFNode, nncf_graph: NNCFGraph) -> bool:
-        assert node.metatype in self.mat_mul_metatypes
-        return len(get_weight_tensor_port_ids(node, nncf_graph)) > 0
+        return node.metatype in self.mat_mul_metatypes and len(get_weight_tensor_port_ids(node, nncf_graph)) > 0
diff --git a/nncf/quantization/algorithms/min_max/torch_fx_backend.py b/nncf/quantization/algorithms/min_max/torch_fx_backend.py
@@ -360,5 +360,4 @@ def get_weight_nodes(self, nncf_graph: NNCFGraph) -> List[NNCFNode]:
         return weight_nodes
 
     def is_matmul_with_constant(self, node: NNCFNode, nncf_graph: NNCFGraph) -> bool:
-        assert node.metatype in self.mat_mul_metatypes
-        return len(get_weight_tensor_port_ids(node, nncf_graph)) > 0
+        return node.metatype in self.mat_mul_metatypes and len(get_weight_tensor_port_ids(node, nncf_graph)) > 0