Determine fine grained memory availability at RCCL bootstrapping (#471)

Tento commit je obsažen v:
Wenkai Du
2021-11-19 08:12:53 -08:00
odevzdal GitHub
rodič 7b8b54955b
revize e9bf01fb7e
5 změnil soubory, kde provedl 14 přidání a 24 odebrání
-10
Zobrazit soubor
@@ -89,16 +89,6 @@ static ncclResult_t ncclCudaMemcpy(T* dst, T* src, size_t nelem) {
return ncclSuccess;
}
static bool hasFineGrainVramPcie() {
int *ptr;
if (hipExtMallocWithFlags((void**)&ptr, sizeof(int), hipDeviceMallocFinegrained) == hipSuccess) {
CUDACHECK(hipFree(ptr));
return true;
}
else
return false;
}
// Allocate memory to be potentially ibv_reg_mr'd. This needs to be
// allocated on separate pages as those pages will be marked DONTFORK
// and if they are shared, that could cause a crash in a child process