diff --git a/src/memory.c b/src/memory.c index f121726f77..964becdc0a 100644 --- a/src/memory.c +++ b/src/memory.c @@ -161,8 +161,8 @@ HSAKMT_STATUS HSAKMTAPI hsaKmtAllocMemory(HSAuint32 PreferredNode, if (zfb_support && gpu_id && MemFlags.ui32.NonPaged == 1) MemFlags.ui32.CoarseGrain = 1; - *MemoryAddress = fmm_allocate_host(gpu_id, PreferredNode, *MemoryAddress, - SizeInBytes, MemFlags); + *MemoryAddress = fmm_allocate_host(gpu_id, MemFlags.ui32.GTTAccess ? 0 : PreferredNode, + *MemoryAddress, SizeInBytes, MemFlags); if (!(*MemoryAddress)) { pr_err("[%s] failed to allocate %lu bytes from host\n",