From efd0158724bbdeb6b86dc17d8e2e81206564dd64 Mon Sep 17 00:00:00 2001
From: tiancyin <tianci.yin@amd.com>
Date: Fri, 13 Sep 2024 10:06:46 +0800
Subject: [PATCH] wsl/hsakmt: use GpuMemory object address as va in physical
 memory handle case

Signed-off-by: tiancyin <tianci.yin@amd.com>
---
 memory.cpp | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/memory.cpp b/memory.cpp
index 814df1c42b..8c47bfb282 100644
--- a/memory.cpp
+++ b/memory.cpp
@@ -326,7 +326,7 @@ HSAKMT_STATUS HSAKMTAPI hsaKmtImportDMABufHandle(int DMABufFd,
 
   auto code = dev->CreateGpuMemory(create_info, &gpu_mem);
   if (code == ErrorCode::Success) {
-    *MemoryAddress = reinterpret_cast<void *>(gpu_mem->GpuAddress());
+    *MemoryAddress = reinterpret_cast<void *>(gpu_mem);
     std::lock_guard<std::mutex> gard(*allocation_map_lock_);
     /*
      * the gpu_mem->Flags() need convert back from GpuMemoryCreateFlags to