Adding explicit request for coarse-grained host memory due to changes in HipHostMalloc (#517)

[ROCm/rccl commit: 2d558c9abc]
This commit is contained in:
gilbertlee-amd
2022-03-25 13:05:07 -06:00
committed by GitHub
parent 95c6476678
commit 4c32c51772
@@ -884,7 +884,7 @@ void AllocateMemory(MemType memType, int devIndex, size_t numBytes, float** memP
}
// Allocate host-pinned memory (should respect NUMA mem policy)
HIP_CALL(hipHostMalloc((void **)memPtr, numBytes, hipHostMallocNumaUser));
HIP_CALL(hipHostMalloc((void **)memPtr, numBytes, hipHostMallocNumaUser | hipHostMallocNonCoherent));
// Check that the allocated pages are actually on the correct NUMA node
CheckPages((char*)*memPtr, numBytes, numaIdx);