Skip to content

Commit f4ef390

Browse files
Apply suggestions from code review
Co-authored-by: Alexander Dokuchaev <[email protected]>
1 parent 0806286 commit f4ef390

File tree

3 files changed

+11
-11
lines changed

3 files changed

+11
-11
lines changed

src/nncf/common/graph/graph.py

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -812,16 +812,16 @@ def find_matching_subgraphs(self, patterns: GraphPattern, strict: bool = True) -
812812
return output
813813

814814

815-
def get_ignored_names_by_ignored_patterns(nncf_graph: NNCFGraph, ignored_patterns: GraphPattern) -> set[str]:
815+
def get_node_names_matching_graph_pattern(nncf_graph: NNCFGraph, graph_pattern: GraphPattern) -> set[str]:
816816
"""
817-
Returns node names matched ignored_patterns.
817+
Returns the names of nodes in the given NNCFGraph that match the specified graph pattern.
818818
819-
:param nncf_graph: NNCFGraph instance.
820-
:param ignored_patterns: Ignored patterns.
821-
:return: A set of NNCF node names from the given NNCFGraph matched ignored_patterns.
819+
:param nncf_graph: An instance of NNCFGraph to search for matching subgraphs.
820+
:param graph_pattern: A GraphPattern instance used to identify matching subgraphs.
821+
:return: A set of node names from the NNCFGraph that match the given graph pattern.
822822
"""
823823
nncf_node_names = set()
824-
for subgraph in nncf_graph.find_matching_subgraphs(ignored_patterns, strict=False):
824+
for subgraph in nncf_graph.find_matching_subgraphs(graph_pattern, strict=False):
825825
for nncf_node in subgraph:
826826
nncf_node_names.add(nncf_node.node_name)
827827
return nncf_node_names

src/nncf/quantization/algorithms/min_max/algorithm.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -22,7 +22,7 @@
2222
from nncf.common.factory import ModelTransformerFactory
2323
from nncf.common.graph.graph import NNCFGraph
2424
from nncf.common.graph.graph import NNCFNode
25-
from nncf.common.graph.graph import get_ignored_names_by_ignored_patterns
25+
from nncf.common.graph.graph import get_node_names_matching_graph_pattern
2626
from nncf.common.graph.operator_metatypes import OperatorMetatype
2727
from nncf.common.graph.patterns import GraphPattern
2828
from nncf.common.graph.patterns.manager import PatternsManager
@@ -586,7 +586,7 @@ def _get_ignored_names(
586586
user_ignored_names = get_ignored_node_names_from_ignored_scope(
587587
self._ignored_scope, nncf_graph, strict=self._ignored_scope.validate
588588
)
589-
autogenerated_ignored_names = get_ignored_names_by_ignored_patterns(inference_nncf_graph, ignored_patterns)
589+
autogenerated_ignored_names = get_node_names_matching_graph_pattern(inference_nncf_graph, ignored_patterns)
590590
autogenerated_ignored_names |= self._backend_entity.get_ignored_names_by_layer_attributes(inference_nncf_graph)
591591
autogenerated_ignored_names |= self._get_ignored_names_by_algorithm(inference_nncf_graph)
592592
ignored_names = {name: IgnoreReason.AUTOGENERATED for name in autogenerated_ignored_names}

src/nncf/quantization/algorithms/weight_compression/algorithm.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -20,7 +20,7 @@
2020
from nncf.common.factory import StatisticsAggregatorFactory
2121
from nncf.common.graph.graph import NNCFGraph
2222
from nncf.common.graph.graph import NNCFNode
23-
from nncf.common.graph.graph import get_ignored_names_by_ignored_patterns
23+
from nncf.common.graph.graph import get_node_names_matching_graph_pattern
2424
from nncf.common.graph.transformations.commands import TargetType
2525
from nncf.common.logging import nncf_logger
2626
from nncf.common.logging.track_progress import track
@@ -385,10 +385,10 @@ def get_nodes_to_compress(self, nncf_graph: NNCFGraph) -> list[NNCFNode]:
385385
self._ignored_scope, nncf_graph, strict=self._ignored_scope.validate
386386
)
387387

388-
ignored_patterns_names = get_ignored_names_by_ignored_patterns(
388+
autogenerated_ignored_names = get_node_names_matching_graph_pattern(
389389
nncf_graph, self._backend_entity.get_ignored_patterns()
390390
)
391-
ignored_names = ignored_names.union(ignored_patterns_names)
391+
ignored_names = ignored_names.union(autogenerated_ignored_names)
392392

393393
for node in nncf_graph.topological_sort():
394394
is_node_with_weights = self._backend_entity.is_node_with_weights(node, nncf_graph)

0 commit comments

Comments
 (0)