tree 12f7632f9af9962e01056a688955d15617301c81
parent 05304f59f055283d7d0d0f0002c0a1682c0f2cb3
author David Berard <dberard@fb.com> 1724172281 -0700
committer PyTorch MergeBot <pytorchmergebot@users.noreply.github.com> 1724265654 +0000

C++ network flow implementation in c10 (#132188)

The functorch partitioners use network flow to split the joint graph into a forward and backward graph. Internally, we've found that upgrading to networkx 2.8.8 (from 2.5) results in some hard-to-debug failures (internal reference: https://fburl.com/workplace/jrqwagdm). And I'm told that there's interest to remove the python dependency.

So this PR introduces a C++ implementation that mirrors the API provided by networkx. We'll need to add python bindings and do some additional testing to verify correctness.

Differential Revision: [D61550977](https://our.internmc.facebook.com/intern/diff/D61550977)
Pull Request resolved: https://github.com/pytorch/pytorch/pull/132188
Approved by: https://github.com/Chillee