Google Git
Sign in
android / platform / external / pytorch / 89ea6acde2 / . / torch / lib / c10d
tree: 77fab6e0cfb3fb9ad911f4ce2d3a69b0fc24509b [path history] [tgz]
  1. bin/
  2. example/
  3. private/
  4. test/
  5. CMakeLists.txt
  6. CUDAUtils.cpp
  7. CUDAUtils.hpp
  8. FileStore.cpp
  9. FileStore.hpp
  10. ProcessGroup.cpp
  11. ProcessGroup.hpp
  12. ProcessGroupGloo.cpp
  13. ProcessGroupGloo.hpp
  14. ProcessGroupMPI.cpp
  15. ProcessGroupMPI.hpp
  16. README.md
  17. Store.cpp
  18. Store.hpp
  19. TCPStore.cpp
  20. TCPStore.hpp
  21. Types.hpp
  22. Utils.cpp
  23. 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