4 Commits

Author	SHA1	Message	Date
PyTorch MergeBot	4035a53cca	Revert "Recursively print graph module and its submodule (#81080)" ... This reverts commit fe7262329c. Reverted https://github.com/pytorch/pytorch/pull/81080 on behalf of https://github.com/DanilBaibak due to Break internal build	2022-07-18 14:46:26 +00:00
Sherlock Huang	fe7262329c	Recursively print graph module and its submodule (#81080) ... Pull Request resolved: https://github.com/pytorch/pytorch/pull/81080 Approved by: https://github.com/ezyang	2022-07-18 01:19:03 +00:00
Sherlock Huang	ac5a94789f	Refactor lift_subgraph_as_module as a fx.passes.util function (#80292) ... lift_subgraph_as_module can be shared between fuser_utils.py and spliter_utils.py Pull Request resolved: https://github.com/pytorch/pytorch/pull/80292 Approved by: https://github.com/jjsjann123, https://github.com/842974287	2022-06-29 22:35:39 +00:00
Sherlock Huang	752c06e0e1	FX graph partitioner and fuser (#79439) ... This PR introduces two components. CapabilityBasedPartitioner for FX graph: given a list of supported operators, this partitioner tries to forms the largest subgraphs that only contain the supported ops. Fuser utility: given a list of nodes in FX graph, it lifts them as a sub-GraphModule in the original graph. Pull Request resolved: https://github.com/pytorch/pytorch/pull/79439 Approved by: https://github.com/jjsjann123, https://github.com/davidberard98	2022-06-24 18:49:37 +00:00