Enable GPU direct RDMA read from GPU

Tá an tiomantas seo le fáil i:
Wenkai Du tiomanta 2021-02-03 02:48:30 +00:00
1 tuismitheoir 01a998b17c
tiomantas 5f97122442
4 comhad athraithe
+27 -4

Níor tháinig aon chomhaid le do chuardach

-4
Féach ar an gComhad
@@ -316,9 +316,6 @@ ncclResult_t ncclTopoCheckGdr(struct ncclTopoSystem* system, int64_t busId, int
if (read) { // For reads (sends) only enable under certain conditions
int gdrReadParam = ncclParamNetGdrRead();
if (gdrReadParam == 0) return ncclSuccess;
#if defined(__HIP_PLATFORM_HCC__) || defined(__HCC__) || defined(__HIPCC__)
return ncclSuccess;
#else
if (gdrReadParam < 0) {
int nvlink = 0;
// Since we don't know whether there are other communicators,
@@ -333,7 +330,6 @@ ncclResult_t ncclTopoCheckGdr(struct ncclTopoSystem* system, int64_t busId, int
}
if (!nvlink) return ncclSuccess;
}
#endif
}
// Check if we are close enough that it makes sense to enable GDR