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

[ROCm/rccl commit: 2d558c9abc]
Этот коммит содержится в:
gilbertlee-amd
2022-03-25 13:05:07 -06:00
коммит произвёл GitHub
родитель 95c6476678
Коммит 4c32c51772
+1 -1
Просмотреть файл
@@ -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);