libhsakmt: Use node_id in topology_is_svm_needed()

Signed-off-by: Graham Sider <Graham.Sider@amd.com> Change-Id: I7773ccb959e60e43b816477c5659f1fe835ecbbd
2021-07-16 16:19:02 -04:00
parent 2f261a8dab
commit bc4aca039c
5 changed files with 12 additions and 13 deletions
@@ -171,7 +171,8 @@ HSAKMT_STATUS HSAKMTAPI hsaKmtAllocMemory(HSAuint32 PreferredNode,
 	}

 	/* GPU allocated VRAM */
-	*MemoryAddress = fmm_allocate_device(gpu_id, *MemoryAddress, SizeInBytes, MemFlags);
+	*MemoryAddress = fmm_allocate_device(gpu_id, PreferredNode, *MemoryAddress,
+					     SizeInBytes, MemFlags);

 	if (!(*MemoryAddress)) {
 		pr_err("[%s] failed to allocate %lu bytes from device\n",