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

This commit is contained in:
Wenkai Du
2021-11-19 08:12:53 -08:00
committed by GitHub
parent 7b8b54955b
commit e9bf01fb7e
5 changed files with 14 additions and 24 deletions
-10
View File
@@ -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