From 82e73c096b8a16abdedd9fad4db5e8625e04e134 Mon Sep 17 00:00:00 2001
From: agunashe <ajay.gunashekar@amd.com>
Date: Wed, 8 Sep 2021 11:13:04 -0700
Subject: [PATCH] SWDEV-301069 - current device id saved while allocating
 memory in VDI

Change-Id: Ic5426895fb1d152e4e6e1baf8b938f35c85fa3d8
---
 rocclr/platform/memory.hpp | 13 +++++++++++++
 1 file changed, 13 insertions(+)

diff --git a/rocclr/platform/memory.hpp b/rocclr/platform/memory.hpp
index 1135fad02a..e27ce7130d 100644
--- a/rocclr/platform/memory.hpp
+++ b/rocclr/platform/memory.hpp
@@ -141,6 +141,11 @@ class Memory : public amd::RuntimeObject {
     kArenaMemoryPtr = 0x2
   };
 
+  struct UserData
+  {
+     int deviceId = 0;
+  };
+
  protected:
   typedef cl_mem_object_type Type;
   typedef cl_mem_flags Flags;
@@ -184,6 +189,9 @@ class Memory : public amd::RuntimeObject {
     uint32_t flagsEx_;
   };
 
+  //! used to save the user data during memory allocation.
+  UserData userData_;
+
  private:
   //! Disable default assignment operator
   Memory& operator=(const Memory&);
@@ -357,6 +365,11 @@ class Memory : public amd::RuntimeObject {
   //! Returns the base device memory object for possible P2P access
   device::Memory* BaseP2PMemory() const { return deviceMemories_[0].value_; }
   device::Memory* svmBase() const { return svmBase_; }  //!< Returns SVM base for MGPU case
+
+
+  //!save the user data during memory allocation
+  UserData& getUserData() { return userData_; }
+
 };
 
 //! Buffers are a specialization of memory. Just a wrapper, really,