From 96cd9f7b769305d04c23cbe9be18fa1a0b49767b Mon Sep 17 00:00:00 2001
From: victzhan <victzhan@amd.com>
Date: Tue, 17 Jan 2023 00:47:28 -0500
Subject: [PATCH] SWDEV-345751 - Added parameter check for clEnqueueWriteImage

Change-Id: I9606ea1c3698d009f23b7da673c02fa05b009dbc
---
 opencl/amdocl/cl_memobj.cpp | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/opencl/amdocl/cl_memobj.cpp b/opencl/amdocl/cl_memobj.cpp
index 33c29b89ad..6dceea6186 100644
--- a/opencl/amdocl/cl_memobj.cpp
+++ b/opencl/amdocl/cl_memobj.cpp
@@ -2368,6 +2368,10 @@ RUNTIME_ENTRY(cl_int, clEnqueueWriteImage,
     return CL_INVALID_OPERATION;
   }
 
+  if (dstImage->getDims() == 2 && origin[2] != 0) {
+    return CL_INVALID_VALUE;
+  }
+
   amd::HostQueue* queue = as_amd(command_queue)->asHostQueue();
   if (NULL == queue) {
     return CL_INVALID_COMMAND_QUEUE;