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