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

This commit is contained in:
Wenkai Du
2021-03-05 19:59:41 -08:00
zatwierdzone przez GitHub
rodzic f4a9b9acba
commit c018edf0f2
14 zmienionych plików z 163 dodań i 44 usunięć
+3
Wyświetl plik
@@ -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)