mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-07 12:21:27 +01:00
268859ce0d
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/19631 ghimport-source-id: edc47e77d6ef03e966944ff98eefc22f2574eeaa Reviewed By: mrshenli Differential Revision: D15110077 Pulled By: mxw fbshipit-source-id: 27a68308ade5ea511e2ea568a071eedb5d21c1ba |
||
|---|---|---|
| .. | ||
| bin | ||
| example | ||
| test | ||
| CMakeLists.txt | ||
| FileStore.cpp | ||
| FileStore.hpp | ||
| NCCLUtils.hpp | ||
| PrefixStore.cpp | ||
| PrefixStore.hpp | ||
| ProcessGroup.cpp | ||
| ProcessGroup.hpp | ||
| ProcessGroupGloo.cpp | ||
| ProcessGroupGloo.hpp | ||
| ProcessGroupMPI.cpp | ||
| ProcessGroupMPI.hpp | ||
| ProcessGroupNCCL.cpp | ||
| ProcessGroupNCCL.hpp | ||
| README.md | ||
| Store.cpp | ||
| Store.hpp | ||
| TCPStore.cpp | ||
| TCPStore.hpp | ||
| Types.hpp | ||
| Utils.cpp | ||
| Utils.hpp | ||
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