Google Git
Sign in
android / platform / external / pytorch / 561bc09026 / . / torch / lib / c10d
tree: 1b99bf7ee06906724735350d1cd7f6b1d0647f16 [path history] [tgz]
  1. bin/
  2. cmake/
  3. example/
  4. test/
  5. CMakeLists.txt
  6. Def.hpp
  7. FileStore.cpp
  8. FileStore.hpp
  9. NCCLUtils.hpp
  10. PrefixStore.cpp
  11. PrefixStore.hpp
  12. ProcessGroup.cpp
  13. ProcessGroup.hpp
  14. ProcessGroupGloo.cpp
  15. ProcessGroupGloo.hpp
  16. ProcessGroupMPI.cpp
  17. ProcessGroupMPI.hpp
  18. ProcessGroupNCCL.cpp
  19. ProcessGroupNCCL.hpp
  20. README.md
  21. Store.cpp
  22. Store.hpp
  23. TCPStore.cpp
  24. TCPStore.hpp
  25. Types.hpp
  26. Utils.cpp
  27. Utils.hpp
torch/lib/c10d/README.md

THD refactor

This is a work in progress. It is separate from the main THD directory to avoid disrupting THD users or have to deal with backwards compat early on. Once this gets to a usable state, we'll add Python bindings and a compat layer.

See https://github.com/pytorch/pytorch/issues/7434 for the main issue.

This tree is intentionally not part of the main build and will be buildable/testable in isolation, as long as ATen is available in <repository root>/torch/lib/tmp_install.

To build and install ATen here, navigate to the root of this repository and run:

tools/build_pytorch_libs.sh --with-cuda ATen