From 96f41fad07a01b04bd3c3ea435d8ed93fee7ac67 Mon Sep 17 00:00:00 2001
From: pghafari <pghafari@amd.com>
Date: Thu, 29 Jun 2023 17:10:30 -0400
Subject: [PATCH] SWDEV-369567 - [ABI Break] updating gridY and Z max dimension
 check

Change-Id: If515fda4938d3c5f455a02b4a65adaad76dc0aed
---
 hipamd/src/hip_module.cpp | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/hipamd/src/hip_module.cpp b/hipamd/src/hip_module.cpp
index 8743beb2a8..eea5efd41b 100644
--- a/hipamd/src/hip_module.cpp
+++ b/hipamd/src/hip_module.cpp
@@ -444,8 +444,8 @@ hipError_t hipModuleLaunchKernel(hipFunction_t f, uint32_t gridDimX, uint32_t gr
   STREAM_CAPTURE(hipModuleLaunchKernel, hStream, f, gridDimX, gridDimY, gridDimZ, blockDimX,
                  blockDimY, blockDimZ, sharedMemBytes, kernelParams, extra);
   if (gridDimX > std::numeric_limits<int32_t>::max() ||
-      gridDimY > std::numeric_limits<int32_t>::max() ||
-      gridDimZ > std::numeric_limits<int32_t>::max()) {
+      gridDimY > std::numeric_limits<int32_t>::max()/1024 ||
+      gridDimZ > std::numeric_limits<int32_t>::max()/1024) {
     HIP_RETURN(hipErrorInvalidValue);
   }
   size_t globalWorkSizeX = static_cast<size_t>(gridDimX) * blockDimX;