Enable local sendrecv over network if GDR is available on all GPUs (#324)

[ROCm/rccl commit: c018edf0f2]
Αυτή η υποβολή περιλαμβάνεται σε:
Wenkai Du authored and GitHub committed 2021-03-05 19:59:41 -08:00
1 γονέας 8d90821062
υποβολή bcf4ecb0e3
14 files changed
+163 -44

No files matched your search

@@ -121,6 +121,9 @@ struct ncclComm {
// Flag to ask NCCL kernels to abort
volatile uint32_t *abortFlag;
// Flags for enable P2P NET
uint32_t *p2pNet;
// Device side of the communicator
struct ncclDevComm *devComm;
// Host copy of the devComm (to free CUDA allocs)
@@ -150,6 +150,8 @@ struct ncclTree {
struct ncclPeer {
struct ncclConnector send;
struct ncclConnector recv;
struct ncclConnector p2pSend;
struct ncclConnector p2pRecv;
};
struct ncclDevComm;
@@ -332,6 +334,9 @@ struct ncclDevComm {
// Channels, device side
struct ncclChannel* channels;
// Flags for enable P2P NET
uint32_t *p2pNet;
#ifdef ENABLE_PROFILING
// Profiling counters
struct ncclProf* devProf;