SWDEV-432599 - Added null checks for rocMemory ptrs in copyMemory

Change-Id: I8ccda2c915dad22c4ea67556dfaab9ba47770069
ROCm · Jan 18, 2024 · 28743d8 · 28743d8
1 parent 0f9f8b5
commit 28743d8
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/rocclr/device/rocm/rocvirtual.cpp b/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;