tree 8e0d1b38bcf7caaf8b0c1916eae6b242a5a76244
parent 86682caf9364348fc8a9d37032c55e6e0c230423
author Sherlock Huang <bahuang@fb.com> 1656096577 +0000
committer PyTorch MergeBot <pytorchmergebot@users.noreply.github.com> 1656096577 +0000

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