From fb2b87db568d2e876ff19862996ffc09466978d6 Mon Sep 17 00:00:00 2001
From: Jaydeep Patel <jaydeepkumar.patel@amd.com>
Date: Wed, 10 Jul 2024 18:07:35 +0000
Subject: [PATCH] SWDEV-471298 - Use same context during child creation as
 parent's context.

Change-Id: I41e534b6194cef9aa8e96b28b8e811906cb362f0
---
 rocclr/device/device.cpp | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/rocclr/device/device.cpp b/rocclr/device/device.cpp
index f4b3fb2a2b..f777a8a03f 100644
--- a/rocclr/device/device.cpp
+++ b/rocclr/device/device.cpp
@@ -430,8 +430,8 @@ amd::Memory* Device::CreateVirtualBuffer(amd::Context& device_context, void* vpt
 
     size_t offset = (reinterpret_cast<address>(vptr)
                      - reinterpret_cast<address>(vaddr_base_obj->getSvmPtr()));
-    vaddr_sub_obj = new (device_context) amd::Buffer(*vaddr_base_obj, CL_MEM_VA_RANGE_AMD, offset,
-                                                     size);
+    Context& ctx = vaddr_base_obj->getContext();
+    vaddr_sub_obj = new (ctx) amd::Buffer(*vaddr_base_obj,CL_MEM_VA_RANGE_AMD, offset, size);
 
     // This curr_mem_obj->create() does not create an actual memory but stores the memory info
     // with given vptr on ROCr backend.