From 8f7eeb8a7461f243480c805b41c2f2d6b019f8aa Mon Sep 17 00:00:00 2001
From: kjayapra-amd <karthik.jayaprakash@amd.com>
Date: Wed, 12 Jul 2023 18:01:51 -0400
Subject: [PATCH] SWDEV-408891 - Only register the host pointer to the current
 device.

Change-Id: If5397f148638e3b7d542b9e6fb8ac749a602b35e


[ROCm/clr commit: 8c7f59663f891269da3c4d32ee80b1208db84dcf]
---
 projects/clr/rocclr/device/rocm/rocmemory.cpp | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/projects/clr/rocclr/device/rocm/rocmemory.cpp b/projects/clr/rocclr/device/rocm/rocmemory.cpp
index 5fdfc0bfff..bb43c7c0a6 100644
--- a/projects/clr/rocclr/device/rocm/rocmemory.cpp
+++ b/projects/clr/rocclr/device/rocm/rocmemory.cpp
@@ -961,8 +961,9 @@ bool Buffer::create(bool alloc_local) {
                                     dev().SystemSegment() :
                                     (dev().SystemCoarseSegment().handle != 0 ?
                                         dev().SystemCoarseSegment() : dev().SystemSegment());
+      hsa_agent_t hsa_agent = dev().getBackendDevice();
       hsa_status_t status = hsa_amd_memory_lock_to_pool(owner()->getHostMem(),
-          owner()->getSize(), nullptr, 0, pool, 0, &deviceMemory_);
+          owner()->getSize(), &hsa_agent, 1, pool, 0, &deviceMemory_);
       ClPrint(amd::LOG_DEBUG, amd::LOG_MEM, "Locking to pool %p, size 0x%zx, HostPtr = %p,"
               " DevPtr = %p", pool, owner()->getSize(), owner()->getHostMem(), deviceMemory_ );
       if (status != HSA_STATUS_SUCCESS) {