Fix producer check

daniil-lyakhov · Jun 29, 2023 · a4f7cab · a4f7cab
1 parent 075e428
commit a4f7cab
Showing 1 changed file with 0 additions and 12 deletions.
diff --git a/nncf/quantization/algorithms/channel_alignment/algorithm.py b/nncf/quantization/algorithms/channel_alignment/algorithm.py
@@ -241,15 +241,6 @@ def _check_consumer_conv_node(self, conv_node: NNCFNode) -> bool:
             return False
         return True
 
-    def _check_producer_conv_node(self, conv_node: NNCFNode):
-        attrs = self._backend_entity.get_conv_layer_attributes(conv_node)
-        if attrs is None:
-            return False
-        # Check groups amount == 1
-        if attrs.groups != 1:
-            return False
-        return True
-
     def _get_target_patterns(self) -> GraphPattern:
         input_attrs = {
             GraphPattern.LABEL_ATTR: "INPUT",
@@ -333,9 +324,6 @@ def _get_node_pairs(self, nncf_graph: NNCFGraph) -> List[Tuple[NNCFNode, Optiona
             else:
                 conv_in, add_in, conv_out = subgraph
 
-            if not self._check_producer_conv_node(conv_in):
-                continue
-
             if not self._check_consumer_conv_node(conv_out):
                 continue