From b72a7113fb9850c59f842343cb4a741bccfe232d Mon Sep 17 00:00:00 2001
From: Ioannis Assiouras <Ioannis.Assiouras@amd.com>
Date: Wed, 15 Nov 2023 13:04:22 +0000
Subject: [PATCH] SWDEV-432599 - Added null checks for rocMemory ptrs in
 copyMemory

Change-Id: I8ccda2c915dad22c4ea67556dfaab9ba47770069


[ROCm/clr commit: 28743d8dbd47423ce5b97e7b6931e3db25f22d68]
---
 projects/clr/rocclr/device/rocm/rocvirtual.cpp | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/projects/clr/rocclr/device/rocm/rocvirtual.cpp b/projects/clr/rocclr/device/rocm/rocvirtual.cpp
index 400b791a81..0046f130b3 100644
--- a/projects/clr/rocclr/device/rocm/rocvirtual.cpp
+++ b/projects/clr/rocclr/device/rocm/rocvirtual.cpp
@@ -1857,7 +1857,10 @@ bool VirtualGPU::copyMemory(cl_command_type type, amd::Memory& srcMem, amd::Memo
                             amd::CopyMetadata copyMetadata) {
   Memory* srcDevMem = dev().getRocMemory(&srcMem);
   Memory* dstDevMem = dev().getRocMemory(&dstMem);
-
+  if (srcDevMem == nullptr || dstDevMem == nullptr) {
+    LogError("submitCopyMemory failed!");
+    return false;
+  }
   // Synchronize source and destination memory
   device::Memory::SyncFlags syncFlags;
   syncFlags.skipEntire_ = entire;