pytorch/torch/fx/passes/utils
Jerry Zhang 7a69e3d30b [fx][subgraph_matcher] Add a matcher that supports name to node map (#110743)
Summary:
We want the matcher to return a name -> node in target graph
so that we can refer to the node by name, this is useful for downstream applications like
quantization.

and also we can use the torch API as source of truth instead of matching aten API directly.

Test Plan:
python test/fx/test_matcher_utils.py

Reviewers:

Subscribers:

Tasks:

Tags:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/110743
Approved by: https://github.com/SherlockNoMad
2023-10-10 22:21:24 +00:00
..
__init__.py An improved version of subgraph matcher (#82090) 2022-08-12 03:32:09 +00:00
common.py Back out "Revert D49107540: [pytorch][PR] split by tag" (#109332) 2023-09-16 05:29:16 +00:00
fuser_utils.py Back out "Revert D49107540: [pytorch][PR] split by tag" (#109332) 2023-09-16 05:29:16 +00:00
matcher_utils.py [fx][subgraph_matcher] Add a matcher that supports name to node map (#110743) 2023-10-10 22:21:24 +00:00
matcher_with_name_node_map_utils.py [fx][subgraph_matcher] Add a matcher that supports name to node map (#110743) 2023-10-10 22:21:24 +00:00
source_matcher_utils.py Replace node.meta source_fn with source_fn_stack (#108595) 2023-09-28 18:18:36 +00:00