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

Цей коміт міститься в:
Wenkai Du
2021-03-05 19:59:41 -08:00
зафіксовано GitHub
джерело f4a9b9acba
коміт c018edf0f2
14 змінених файлів з 163 додано та 44 видалено
+3
Переглянути файл
@@ -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)