From f960433dcd6da76d43dabc3cffaa36d2b285b75c Mon Sep 17 00:00:00 2001
From: "Zhang, Victor" <Victor.Zhang@amd.com>
Date: Wed, 7 May 2025 09:52:09 -0400
Subject: [PATCH] SWDEV-528142 - add error check for KernelParameters::capture
 (#276)

* SWDEV-528142 - add error check for KernelParameters::capture

* Update kernel.cpp

---------

Co-authored-by: victzhan <victzhan@amd.com>
---
 rocclr/platform/kernel.cpp | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/rocclr/platform/kernel.cpp b/rocclr/platform/kernel.cpp
index 2fb08f4e92..9eb744392b 100644
--- a/rocclr/platform/kernel.cpp
+++ b/rocclr/platform/kernel.cpp
@@ -282,6 +282,11 @@ address KernelParameters::capture(device::VirtualDevice& vDev, uint64_t lclMemSi
       } else if (desc.type_ == T_SAMPLER) {
         Sampler* samplerArg = samplerObjects_[desc.info_.arrayIndex_];
         if (samplerArg != nullptr) {
+          device::Sampler* deviceSampler = samplerArg->getDeviceSampler(device);
+          if (!deviceSampler) {
+            *error = CL_INVALID_CONTEXT;
+            break;
+          }
           samplerArg->retain();
           // todo: It's uint64_t type
           *reinterpret_cast<uintptr_t*>(mem + desc.offset_) = static_cast<uintptr_t>(