Skip to content

Commit 0658dce

Browse files
mcremon-metafacebook-github-bot
authored andcommitted
Print overload name in print_ops_info (#5404)
Summary: Pull Request resolved: #5404 As titled. This is helpful when the aten ops have many overloads, e.g. `aten.mean.out` or `aten.mean.dim_out` Reviewed By: dulinriley Differential Revision: D62786631 fbshipit-source-id: d5351aa5729385225d8348baa9aef1eb998153f6
1 parent fe53daf commit 0658dce

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

backends/cadence/aot/utils.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -104,11 +104,11 @@ def get_ops_count(graph_module: torch.fx.GraphModule) -> Dict[str, int]:
104104
):
105105
continue
106106
# If the op is already present, increment the count
107-
if get_edge_overload_packet(node.target).__name__ in freq:
108-
freq[get_edge_overload_packet(node.target).__name__] += 1
107+
if node.target._name in freq:
108+
freq[node.target._name] += 1
109109
# else, add a new entry
110110
else:
111-
freq[get_edge_overload_packet(node.target).__name__] = 1
111+
freq[node.target._name] = 1
112112
return freq
113113

114114

0 commit comments

Comments
 (0)