2.4.8-1
Fix #209: improve socket transport performance
Split transfers over multiple sockets
Launch multiple threads to drive sockets
Detect AWS NICs and set nsockets/nthreads accordingly
[ROCm/rccl commit: 7c72dee660]
This commit is contained in:
@@ -124,14 +124,15 @@ cleanup:
|
||||
}
|
||||
|
||||
ncclResult_t initNet() {
|
||||
// Always initialize sockets as we use it for bootstrap
|
||||
NCCLCHECK(initNet(&ncclNetSocket));
|
||||
// Always initialize bootstrap network
|
||||
NCCLCHECK(bootstrapNetInit());
|
||||
|
||||
NCCLCHECK(initNetPlugin(&ncclNet));
|
||||
if (ncclNet != NULL) return ncclSuccess;
|
||||
if (initNet(&ncclNetIb) == ncclSuccess) {
|
||||
ncclNet = &ncclNetIb;
|
||||
} else {
|
||||
NCCLCHECK(initNet(&ncclNetSocket));
|
||||
ncclNet = &ncclNetSocket;
|
||||
}
|
||||
return ncclSuccess;
|
||||
|
||||
Reference in New Issue
Block a user