David Berard 84b3f1900a 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

2024-08-21 18:40:54 +00:00

..

AbortHandler.h

[9/N] Fixes clang-tidy warnings in c10/util/*.h (#116185)

2023-12-22 09:35:44 +00:00

accumulate.h

[Reland] [11/N] Enable clang-tidy warnings on c10/util/*.h (#116751)

2024-01-08 11:07:58 +00:00

AlignOf.h

[10/N] Fixes clang-tidy warnings in c10/util/*.h (#116326)

2023-12-23 04:59:55 +00:00

ApproximateClock.cpp

[c10] Move profiler clock to libc10 for timestamps (#111972)

2023-10-27 16:18:40 +00:00

ApproximateClock.h

[Reland] [11/N] Enable clang-tidy warnings on c10/util/*.h (#116751)

2024-01-08 11:07:58 +00:00

Array.h

Update dispatch stub to make SDPA routing cleaner (#126832)

2024-05-25 01:40:53 +00:00

ArrayRef.h

[codemod] c10:optional -> std::optional (#126135)

2024-05-14 19:35:51 +00:00

Backtrace.cpp

[1/N] Change #include <c10/util/Optional.h> to #include <optional> (#128301)

2024-07-08 07:03:53 +00:00

Backtrace.h

[caffe2] Make all get_backtrace() implementations lazy (#125750) (#126064)

2024-05-13 20:17:41 +00:00

Bfloat16.cpp

[AOTI] Move c10/util ostream function implementations to their headers (#123847)

2024-04-19 00:51:24 +00:00

BFloat16-inl.h

[torch][take2] Implement BFloat16 __hip_bfloat16 overloads (#132234)

2024-08-01 04:25:46 +00:00

BFloat16-math.h

Remove use of math_compat.h (#116167)

2024-01-19 03:37:55 +00:00

BFloat16.h

[torch][take2] Implement BFloat16 __hip_bfloat16 overloads (#132234)

2024-08-01 04:25:46 +00:00

bit_cast.h

[5/N] Fixes clang-tidy warnings in c10/{core,util}/*.h (#115354)

2023-12-09 17:16:04 +00:00

bits.h

Add various uninterpreted bit tensor data types (try 2) (#95860)

2023-03-04 03:35:59 +00:00

Bitset.h

[10/N] Fixes clang-tidy warnings in c10/util/*.h (#116326)

2023-12-23 04:59:55 +00:00

BUILD.bazel

extract out //c10/util:base library (#70854)

2022-01-26 11:51:45 +00:00

build.bzl

migrate PyTorch to c10::bit_cast (#98418)

2023-04-06 19:38:06 +00:00

C++17.cpp

…

C++17.h

Remove c10::guts::{conjunction,disjunction} (#127726)

2024-06-03 04:06:21 +00:00

CallOnce.h

[10/N] Fixes clang-tidy warnings in c10/util/*.h (#116326)

2023-12-23 04:59:55 +00:00

complex_math.cpp

Remove use of math_compat.h (#116167)

2024-01-19 03:37:55 +00:00

complex_math.h

[codemod][lowrisk] Remove extra semi colon from caffe2/c10/util/Float8_e5m2.h (#115761)

2024-01-04 02:02:26 +00:00

complex_utils.h

Logcumsumexp for CPU (#93153)

2023-01-27 22:29:33 +00:00

complex.h

[C10] Mark Complex::imag as C10_HOST_DEVICE (#116877)

2024-01-05 21:17:05 +00:00

ConstexprCrc.h

[Reland] [11/N] Enable clang-tidy warnings on c10/util/*.h (#116751)

2024-01-08 11:07:58 +00:00

copysign.h

Remove use of math_compat.h (#116167)

2024-01-19 03:37:55 +00:00

DeadlockDetection.cpp

Revert "[Environment Variable][1/N] Use thread-safe env variable API in c10 (#119449)"

2024-04-22 14:44:44 +00:00

DeadlockDetection.h

[9/N] Fixes clang-tidy warnings in c10/util/*.h (#116185)

2023-12-22 09:35:44 +00:00

Deprecated.h

[PyTorch] Autoformat c10 (#56830)

2021-04-30 21:23:28 -07:00

DimVector.h

[9/N] Fixes clang-tidy warnings in c10/util/*.h (#116185)

2023-12-22 09:35:44 +00:00

DynamicCounter.cpp

Fix missing include file (#132647)

2024-08-05 15:49:49 +00:00

DynamicCounter.h

[pytorch][counters] DynamicCounter (#132166)

2024-07-31 19:52:51 +00:00

env.h

[9/N] Replace c10::optional with std::optional (#130674)

2024-07-15 00:48:43 +00:00

Exception.cpp

[codemod][lowrisk] Remove extra semi colon from caffe2/c10/core/SymNodeImpl.h (#123055)

2024-05-14 19:35:29 +00:00

Exception.h

[caffe2] Make all get_backtrace() implementations lazy (#125750) (#126064)

2024-05-13 20:17:41 +00:00

ExclusivelyOwned.h

[Reland] [11/N] Enable clang-tidy warnings on c10/util/*.h (#116751)

2024-01-08 11:07:58 +00:00

ExclusivelyOwnedTensorTraits.h

[10/N] Fixes clang-tidy warnings in c10/util/*.h (#116326)

2023-12-23 04:59:55 +00:00

FbcodeMaps.h

[PyTorch] add c10/util/FbcodeMaps.h (#96359)

2023-03-10 02:18:16 +00:00

flags_use_gflags.cpp

adjust header inclusions in C10 as sugguested by IWYU (#102467)

2023-05-31 19:19:10 +00:00

flags_use_no_gflags.cpp

More tidy fixes (#93069)

2023-01-27 06:40:50 +00:00

Flags.h

[9/N] Fixes clang-tidy warnings in c10/util/*.h (#116185)

2023-12-22 09:35:44 +00:00

flat_hash_map.h

[Reland] [11/N] Enable clang-tidy warnings on c10/util/*.h (#116751)

2024-01-08 11:07:58 +00:00

Float8_e4m3fn-inl.h

Enable clang-tidy on c10/util/Float8*.h (#120573)

2024-06-14 13:47:07 +00:00

Float8_e4m3fn.cpp

[AOTI] Move c10/util ostream function implementations to their headers (#123847)

2024-04-19 00:51:24 +00:00

Float8_e4m3fn.h

Enable clang-tidy on c10/util/Float8*.h (#120573)

2024-06-14 13:47:07 +00:00

Float8_e4m3fnuz-inl.h

additional support for float8_e4m3fnuz and _e5m2fnuz (#115214)

2024-01-22 18:33:41 +00:00

Float8_e4m3fnuz.cpp

Enable clang-tidy on c10/util/Float8*.h (#120573)

2024-06-14 13:47:07 +00:00

Float8_e4m3fnuz.h

Enable clang-tidy on c10/util/Float8*.h (#120573)

2024-06-14 13:47:07 +00:00

Float8_e5m2-inl.h

Enable torch.empty for float8 dtypes + deterministic mode + cpu (#128744)

2024-06-15 02:05:30 +00:00

Float8_e5m2.cpp

Enable clang-tidy on c10/util/Float8*.h (#120573)

2024-06-14 13:47:07 +00:00

Float8_e5m2.h

[AOTI] Move c10/util ostream function implementations to their headers (#123847)

2024-04-19 00:51:24 +00:00

Float8_e5m2fnuz-inl.h

Enable torch.empty for float8 dtypes + deterministic mode + cpu (#128744)

2024-06-15 02:05:30 +00:00

Float8_e5m2fnuz.cpp

Enable clang-tidy on c10/util/Float8*.h (#120573)

2024-06-14 13:47:07 +00:00

Float8_e5m2fnuz.h

Eliminate some C++11 checks (#126308)

2024-05-16 22:37:45 +00:00

Float8_fnuz_cvt.h

Build SYCL kernels for ATen XPU ops on Native Windows (take 2) (#127390)

2024-06-06 01:41:06 +00:00

floating_point_utils.h

[10/N] Fixes clang-tidy warnings in c10/util/*.h (#116326)

2023-12-23 04:59:55 +00:00

FunctionRef.h

[10/N] Fixes clang-tidy warnings in c10/util/*.h (#116326)

2023-12-23 04:59:55 +00:00

generic_math.h

Fix c10::div_floor_floating compile error (#115647)

2023-12-20 00:09:01 +00:00

Half-inl.h

[PyTorch] Half: don't disable direct conversion to/from float on mobile (#130465)

2024-07-12 19:46:30 +00:00

Half.cpp

[AOTI] Move c10/util ostream function implementations to their headers (#123847)

2024-04-19 00:51:24 +00:00

Half.h

[1/N] Fix Wunused-parameter warnings (#130924)

2024-07-19 06:14:51 +00:00

hash.h

improve CUDACachingAllocator lock contention (#118550)

2024-02-04 16:45:25 +00:00

IdWrapper.h

[Reland] [11/N] Enable clang-tidy warnings on c10/util/*.h (#116751)

2024-01-08 11:07:58 +00:00

int128.cpp

increase clang-tidy coverage to more c10 source files (#102902)

2023-06-04 06:33:01 +00:00

int128.h

[1/N] Change static functions in headers to inline (#127727)

2024-06-03 04:34:36 +00:00

intrusive_ptr.cpp

…

intrusive_ptr.h

Changes to support clang-19 (#131905)

2024-07-29 12:38:23 +00:00

irange.h

[9/N] Fixes clang-tidy warnings in c10/util/*.h (#116185)

2023-12-22 09:35:44 +00:00

Lazy.h

[caffe2] Lazily symbolize backtrace in c10::Error (#125787)

2024-05-09 01:46:57 +00:00

LeftRight.cpp

…

LeftRight.h

[9/N] Fixes clang-tidy warnings in c10/util/*.h (#116185)

2023-12-22 09:35:44 +00:00

llvmMathExtras.h

[Reland] [11/N] Enable clang-tidy warnings on c10/util/*.h (#116751)

2024-01-08 11:07:58 +00:00

Load.h

[2/N] Fixes clang-tidy warnings in header files (#113727)

2023-11-16 13:21:15 +00:00

logging_is_google_glog.h

Fix TORCH_CHECK macros when glog is used (#83216)

2022-08-11 15:27:45 +00:00

logging_is_not_google_glog.h

[1/N] Fix Wunused-parameter warnings (#130924)

2024-07-19 06:14:51 +00:00

Logging.cpp

[torchscript] Add a sampled logging integration point. (#133484)

2024-08-19 18:04:45 +00:00

Logging.h

[torchscript] Add a sampled logging integration point. (#133484)

2024-08-19 18:04:45 +00:00

MathConstants.cpp

Use c10 math constants consistently in Math.h (#91967)

2023-01-12 11:53:52 +00:00

MathConstants.h

Use c10 math constants consistently in Math.h (#91967)

2023-01-12 11:53:52 +00:00

MaybeOwned.h

[Reland] [11/N] Enable clang-tidy warnings on c10/util/*.h (#116751)

2024-01-08 11:07:58 +00:00

Metaprogramming.cpp

…

Metaprogramming.h

[10/N] Fixes clang-tidy warnings in c10/util/*.h (#116326)

2023-12-23 04:59:55 +00:00

NetworkFlow.cpp

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

2024-08-21 18:40:54 +00:00

NetworkFlow.h

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

2024-08-21 18:40:54 +00:00

numa.cpp

Enable misc clang-tidy checks (#110283)

2023-09-30 10:39:52 +00:00

numa.h

[9/N] Fixes clang-tidy warnings in c10/util/*.h (#116185)

2023-12-22 09:35:44 +00:00

Optional.cpp

[PyTorch] Redirect c10::optional to std::optional (#101995)

2023-11-30 02:46:41 +00:00

Optional.h

[12/N] Use std::optional (#132361)

2024-08-02 13:46:46 +00:00

OptionalArrayRef.h

[1/N] Change #include <c10/util/Optional.h> to #include <optional> (#128301)

2024-07-08 07:03:53 +00:00

order_preserving_flat_hash_map.h

[Reland] [11/N] Enable clang-tidy warnings on c10/util/*.h (#116751)

2024-01-08 11:07:58 +00:00

overloaded.h

[10/N] Fixes clang-tidy warnings in c10/util/*.h (#116326)

2023-12-23 04:59:55 +00:00

ParallelGuard.cpp

Avoid COW materialization in at::parallel_for/parallel_reduce (#120455)

2024-03-01 05:05:28 +00:00

ParallelGuard.h

Avoid COW materialization in at::parallel_for/parallel_reduce (#120455)

2024-03-01 05:05:28 +00:00

python_stub.h

…

qint8.h

[pytorch] add missing host-device attribute to fix clang build (#37358)

2020-04-27 18:24:20 -07:00

qint32.h

[pytorch] add missing host-device attribute to fix clang build (#37358)

2020-04-27 18:24:20 -07:00

quint2x4.h

[oss][pytorch] Add quint2x4 dtype (#65545)

2021-10-06 14:22:00 -07:00

quint4x2.h

[PyTorch] Autoformat c10 (#56830)

2021-04-30 21:23:28 -07:00

quint8.h

Fix misleading doc string in quint8.h (#48418)

2020-11-25 20:48:39 -08:00

Registry.h

[9/N] Fixes clang-tidy warnings in c10/util/*.h (#116185)

2023-12-22 09:35:44 +00:00

safe_numerics.h

[9/N] Fixes clang-tidy warnings in c10/util/*.h (#116185)

2023-12-22 09:35:44 +00:00

ScopeExit.h

[Reland] [11/N] Enable clang-tidy warnings on c10/util/*.h (#116751)

2024-01-08 11:07:58 +00:00

signal_handler.cpp

use condition_variable and wait_until in nccl dump on timeout (#120544)

2024-02-28 00:06:08 +00:00

signal_handler.h

use condition_variable and wait_until in nccl dump on timeout (#120544)

2024-02-28 00:06:08 +00:00

SmallBuffer.h

[Clang-tidy header][21/N] Fix clang-tidy warnings in aten/src/ATEN/*.{cpp,h} (#120763)

2024-03-03 23:18:43 +00:00

SmallVector.cpp

Fix warnings in SmallVector (#127250)

2024-05-30 21:13:20 +00:00

SmallVector.h

Fix warnings in SmallVector (#127250)

2024-05-30 21:13:20 +00:00

sparse_bitset.h

[10/N] Fixes clang-tidy warnings in c10/util/*.h (#116326)

2023-12-23 04:59:55 +00:00

ssize.h

[10/N] Fixes clang-tidy warnings in c10/util/*.h (#116326)

2023-12-23 04:59:55 +00:00

static_tracepoint_elfx86.h

[9/N] Fixes clang-tidy warnings in c10/util/*.h (#116185)

2023-12-22 09:35:44 +00:00

static_tracepoint.h

[caffe2][SDT] Check whether TORCH_DISABLE_SDT macro is defined before referencing it (#107149)

2023-08-16 19:52:12 +00:00

strides.h

[1/N] Change static functions in headers to inline (#127727)

2024-06-03 04:34:36 +00:00

string_utils.h

[Reland] [11/N] Enable clang-tidy warnings on c10/util/*.h (#116751)

2024-01-08 11:07:58 +00:00

string_view.h

Revert "Simplify c10::string_view (#130009)"

2024-07-12 00:46:49 +00:00

StringUtil.cpp

Ack codecvt_utf8_utf16 as a deprecated func in C++17 (#127659)

2024-06-01 04:31:39 +00:00

StringUtil.h

[2/N] Change static functions in headers to inline (#127764)

2024-06-04 00:49:04 +00:00

strong_type.h

STRONG_CONSTEXPR -> constexpr (#125872)

2024-05-13 14:07:26 +00:00

Synchronized.h

[9/N] Fixes clang-tidy warnings in c10/util/*.h (#116185)

2023-12-22 09:35:44 +00:00

tempfile.cpp

Revert "Replace manual parsing of "TMPDIR", "TMP", "TEMP" and "TEMPDIR" with std::filesystem::temp_directory_path() (#130842)"

2024-07-23 17:15:06 +00:00

tempfile.h

[9/N] Fixes clang-tidy warnings in c10/util/*.h (#116185)

2023-12-22 09:35:44 +00:00

thread_name.cpp

Add missing #include <array> to thread_name.cpp (#128664)

2024-06-14 07:49:09 +00:00

thread_name.h

expose set_thread_name to Python and set thread names (#128448)

2024-06-13 16:38:23 +00:00

ThreadLocal.h

Include Macros.h in ThreadLocal

2021-05-27 20:47:56 -07:00

ThreadLocalDebugInfo.cpp

cleanup unused include (#93359)

2023-02-04 02:15:50 +00:00

ThreadLocalDebugInfo.h

[9/N] Fixes clang-tidy warnings in c10/util/*.h (#116185)

2023-12-22 09:35:44 +00:00

Type_demangle.cpp

[Reland 1] Cleanup header inclusions in torch_cpu by iwyu (#112311)

2023-11-19 04:06:36 +00:00

Type_no_demangle.cpp

[2/N] Cleanup header inclusions in torch_cpu by iwyu (#109964)

2023-11-19 20:56:32 +00:00

Type.h

[T153220354] Fix header inclusions in c10 (#1541) (#101846)

2023-05-20 19:35:14 +00:00

TypeCast.cpp

[codemod] Add [[noreturn]] to 2 files inc caffe2/c10/util/TypeCast.cpp (#129575)

2024-06-27 17:23:22 +00:00

TypeCast.h

[codemod] Add [[noreturn]] to 2 files inc caffe2/c10/util/TypeCast.cpp (#129575)

2024-06-27 17:23:22 +00:00

typeid.cpp

[PyTorch] Further reduce cost of TypeMeta::_typeMetaData (by 10x!) (#98105)

2023-04-12 17:44:48 +00:00

typeid.h

Add unsigned integer dtypes to PyTorch (#116594)

2024-01-07 07:40:49 +00:00

TypeIndex.h

Revert "Simplify c10::string_view (#130009)"

2024-07-12 00:46:49 +00:00

TypeList.cpp

…

TypeList.h

Remove unused c10/util/C++17.h inclusion and outdated checks (#120149)

2024-02-17 14:28:17 +00:00

TypeSafeSignMath.h

[2/N] Change static functions in headers to inline (#127764)

2024-06-04 00:49:04 +00:00

TypeTraits.cpp

…

TypeTraits.h

[9/N] Fixes clang-tidy warnings in c10/util/*.h (#116185)

2023-12-22 09:35:44 +00:00

Unicode.cpp

[PyTorch] Autoformat c10 (#56830)

2021-04-30 21:23:28 -07:00

Unicode.h

[PyTorch] Autoformat c10 (#56830)

2021-04-30 21:23:28 -07:00

UniqueVoidPtr.cpp

Enable nested namespace check in clang-tidy (#118506)

2024-01-31 00:32:35 +00:00

UniqueVoidPtr.h

[9/N] Fixes clang-tidy warnings in c10/util/*.h (#116185)

2023-12-22 09:35:44 +00:00

Unroll.h

add int4 packed gemm support on CPU device (#117475)

2024-03-06 16:25:53 +00:00

WaitCounter.cpp

[easy][pytorch][counters] Move WaitCounter in c10/util (#131021)

2024-07-24 18:38:33 +00:00

WaitCounter.h

[pytorch][counters] Pybind for WaitCounter (#132357)

2024-08-02 16:08:10 +00:00

win32-headers.h

[PyTorch] Autoformat c10 (#56830)

2021-04-30 21:23:28 -07:00