Skip to content

Filter types with hexadecimal in NNCFGraph for analysis #63

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
daniil-lyakhov wants to merge 2 commits into
base: ad/torch280
Choose a base branch
from
Open

Filter types with hexadecimal in NNCFGraph for analysis #63

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
2 commits
Select commit Hold shift + click to select a range
08538a4
Filter types with hexadecimal in NNCFGraph for analysis
daniil-lyakhov Aug 11, 2025
a3d5fa1
Code comment
daniil-lyakhov Aug 11, 2025
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
6 changes: 5 additions & 1 deletion src/nncf/common/graph/graph.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -622,7 +622,11 @@ def get_graph_for_structure_analysis(self, extended: bool = False) -> nx.DiGraph
"""
out_graph = nx.DiGraph()
for node_name, node in self._nx_graph.nodes.items():
attrs_node = {"id": str(node[NNCFNode.ID_NODE_ATTR]), "type": node[NNCFNode.NODE_TYPE_ATTR]}
attrs_node = {"id": str(node[NNCFNode.ID_NODE_ATTR])}
# Filter types which have a reference to a memory in its name.
# Relevant for torchFX dynamic graphs since torch==2.8.0
if "0x" not in node[NNCFNode.NODE_TYPE_ATTR]:
attrs_node["type"] = node[NNCFNode.NODE_TYPE_ATTR]
for attr in ["color", "label", "style"]:
if attr in node:
attrs_node[attr] = node[attr]
Expand Down
6 changes: 3 additions & 3 deletions tests/torch2/data/fx/dynamic_shapes/post_quantization_compressed/swin_v2_t.dot
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -252,7 +252,7 @@ strict digraph {
"250 pow_2" [id=250, type=<built-in function pow>];
"251 mul_48" [id=251, type=<built-in function mul>];
"252 eq_74" [id=252, type=<built-in function eq>];
"253 sym_not" [id=253, type=<function sym_not at 0x00000179AB043560>];
"253 sym_not" [id=253];
"254 _assert_scalar_default_5" [id=254, type="_assert_scalar"];
"255 mod_58" [id=255, type=<built-in function mod>];
"256 add_45" [id=256, type=<built-in function add>];
Expand Down Expand Up @@ -309,7 +309,7 @@ strict digraph {
"307 pow_4" [id=307, type=<built-in function pow>];
"308 mul_61" [id=308, type=<built-in function mul>];
"309 eq_80" [id=309, type=<built-in function eq>];
"310 sym_not_1" [id=310, type=<function sym_not at 0x00000179AB043560>];
"310 sym_not_1" [id=310];
"311 _assert_scalar_default_13" [id=311, type="_assert_scalar"];
"312 mod_64" [id=312, type=<built-in function mod>];
"313 add_55" [id=313, type=<built-in function add>];
Expand Down Expand Up @@ -366,7 +366,7 @@ strict digraph {
"364 pow_6" [id=364, type=<built-in function pow>];
"365 mul_74" [id=365, type=<built-in function mul>];
"366 eq_86" [id=366, type=<built-in function eq>];
"367 sym_not_2" [id=367, type=<function sym_not at 0x00000179AB043560>];
"367 sym_not_2" [id=367];
"368 _assert_scalar_default_21" [id=368, type="_assert_scalar"];
"369 mod_70" [id=369, type=<built-in function mod>];
"370 add_65" [id=370, type=<built-in function add>];
Expand Down
6 changes: 3 additions & 3 deletions tests/torch2/data/fx/dynamic_shapes/quantized/swin_v2_t.dot
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -252,7 +252,7 @@ strict digraph {
"250 pow_2" [id=250, type=<built-in function pow>];
"251 mul_48" [id=251, type=<built-in function mul>];
"252 eq_74" [id=252, type=<built-in function eq>];
"253 sym_not" [id=253, type=<function sym_not at 0x0000021654883560>];
"253 sym_not" [id=253];
"254 _assert_scalar_default_5" [id=254, type="_assert_scalar"];
"255 mod_58" [id=255, type=<built-in function mod>];
"256 add_45" [id=256, type=<built-in function add>];
Expand Down Expand Up @@ -309,7 +309,7 @@ strict digraph {
"307 pow_4" [id=307, type=<built-in function pow>];
"308 mul_61" [id=308, type=<built-in function mul>];
"309 eq_80" [id=309, type=<built-in function eq>];
"310 sym_not_1" [id=310, type=<function sym_not at 0x0000021654883560>];
"310 sym_not_1" [id=310];
"311 _assert_scalar_default_13" [id=311, type="_assert_scalar"];
"312 mod_64" [id=312, type=<built-in function mod>];
"313 add_55" [id=313, type=<built-in function add>];
Expand Down Expand Up @@ -366,7 +366,7 @@ strict digraph {
"364 pow_6" [id=364, type=<built-in function pow>];
"365 mul_74" [id=365, type=<built-in function mul>];
"366 eq_86" [id=366, type=<built-in function eq>];
"367 sym_not_2" [id=367, type=<function sym_not at 0x0000021654883560>];
"367 sym_not_2" [id=367];
"368 _assert_scalar_default_21" [id=368, type="_assert_scalar"];
"369 mod_70" [id=369, type=<built-in function mod>];
"370 add_65" [id=370, type=<built-in function add>];
Expand Down