From 432bdd7bf20899d3b2959852bbd46f79d84e60cb Mon Sep 17 00:00:00 2001
From: Rahul Manocha <rmanocha@amd.com>
Date: Thu, 8 Aug 2024 19:48:34 -0700
Subject: [PATCH] SWDEV-474617 SWDEV-464679 - Fix segfault in palvirtual due to
 peer memory access

Change-Id: Ib8b641712d78acf8bc073ca5705dea97af6f944a
---
 rocclr/device/pal/palvirtual.cpp | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/rocclr/device/pal/palvirtual.cpp b/rocclr/device/pal/palvirtual.cpp
index 01f67c8a52..2cedc0bcba 100644
--- a/rocclr/device/pal/palvirtual.cpp
+++ b/rocclr/device/pal/palvirtual.cpp
@@ -1418,6 +1418,11 @@ bool VirtualGPU::copyMemory(cl_command_type type, amd::Memory& srcMem, amd::Memo
   pal::Memory* dstMemory = dev().getGpuMemory(&dstMem);
   pal::Memory* srcMemory = dev().getGpuMemory(&srcMem);
 
+  if(dstMemory == nullptr || srcMemory == nullptr){
+    LogError("submitcopyMemory Failed!");
+    return false;
+  }
+
   // Synchronize source and destination memory
   device::Memory::SyncFlags syncFlags;
   syncFlags.skipEntire_ = entire;