Skip to content

Commit 8f93aa1

Browse files
Code comment
1 parent 08538a4 commit 8f93aa1

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

src/nncf/common/graph/graph.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -623,6 +623,8 @@ def get_graph_for_structure_analysis(self, extended: bool = False) -> nx.DiGraph
623623
out_graph = nx.DiGraph()
624624
for node_name, node in self._nx_graph.nodes.items():
625625
attrs_node = {"id": str(node[NNCFNode.ID_NODE_ATTR])}
626+
# Filter types which have a reference to a memory in the name.
627+
# Actual for torchFX dynamic graph since torch==2.8.0
626628
if "0x" not in node[NNCFNode.NODE_TYPE_ATTR]:
627629
attrs_node["type"] = node[NNCFNode.NODE_TYPE_ATTR]
628630
for attr in ["color", "label", "style"]:

0 commit comments

Comments
 (0)