Skip to content

Commit

Permalink
[NCCL][Test Only] test send/recv on OSS
Browse files Browse the repository at this point in the history
Pull Request resolved: #45140

Test only
ghstack-source-id: 112604790

Differential Revision: [D23844388](https://our.internmc.facebook.com/intern/diff/D23844388/)
  • Loading branch information
mingzhe0908 committed Sep 22, 2020
1 parent af9a570 commit b1fe7cd
Showing 1 changed file with 0 additions and 3 deletions.
3 changes: 0 additions & 3 deletions torch/lib/c10d/NCCLUtils.hpp
Original file line number Diff line number Diff line change
Expand Up @@ -17,8 +17,6 @@
#define ENABLE_NCCL_ERROR_CHECKING
#endif

// Fix build issues with NCCL P2P - until then disable NCCL send/recv.
#if defined(ENABLE_NCCL_A2A) && (ENABLE_NCCL_A2A == 1)
// P2P is enabled only for NCCL versions 2.7+ since ncclSend()
// and ncclRecv() are not supported in earlier versions.
#if defined(NCCL_MAJOR) && (NCCL_MAJOR == 2) && defined(NCCL_MINOR) && \
Expand All @@ -27,7 +25,6 @@
#elif defined(NCCL_MAJOR) && (NCCL_MAJOR >= 3)
#define ENABLE_NCCL_P2P_SUPPORT
#endif
#endif

// Macro to throw on a non-successful NCCL return value.
#define C10D_NCCL_CHECK(cmd) \
Expand Down

0 comments on commit b1fe7cd

Please sign in to comment.