pytorch/test/fx
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
..
named_tup.py
quantization.py
test_common_passes.py
test_cse_pass.py
test_dce_pass.py
test_future.py [BE] Enable ruff's UP rules and autoformat dynamo / functorch and refs (#105432) 2023-07-19 13:48:44 +00:00
test_fx_const_fold.py
test_fx_param_shape_control_flow.py
test_fx_split.py Back out "Revert D49107540: [pytorch][PR] split by tag" (#109332) 2023-09-16 05:29:16 +00:00
test_gradual_type.py [BE] Enable ruff's UP rules and autoformat dynamo / functorch and refs (#105432) 2023-07-19 13:48:44 +00:00
test_matcher_utils.py [fx][subgraph_matcher] Add a matcher that supports name to node map (#110743) 2023-10-10 22:21:24 +00:00
test_pass_infra.py
test_source_matcher_utils.py Change _dynamo.export to be export(f)(*args, **kwargs) (#106109) 2023-07-27 21:41:13 +00:00
test_subgraph_rewriter.py [fx] Better replacements finder in subgraph rewriter (#100556) 2023-05-16 14:08:44 +00:00
test_z3_gradual_types.py