Skip to content

Commit 752cc24

Browse files
Skip nccl_device.h header include for ROCm (causes build failures in theRock nightly builds); changes ported from pytorch#175443
1 parent bb7978b commit 752cc24

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

torch/csrc/distributed/c10d/symm_mem/nccl_dev_cap.hpp

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -10,9 +10,11 @@
1010
#endif
1111

1212
#if NCCL_VERSION_CODE >= NCCL_VERSION(2, 28, 0)
13+
#if !defined(USE_ROCM)
1314
#define NCCL_HAS_SYMMEM_DEVICE_SUPPORT
1415
#include <nccl_device.h>
1516
#endif
17+
#endif
1618

1719
#if NCCL_VERSION_CODE >= NCCL_VERSION(2, 29, 0)
1820
#define NCCL_HAS_ONE_SIDED_API

0 commit comments

Comments
 (0)