SWDEV-267154 - Add checks for grid and block dimensions 0

Change-Id: I54799c0a4a42a91f6226f94a1e3ff7bc79046a98
2021-01-07 05:36:04 -05:00
parent 09c28dcbf4
commit a16d91a708
1 changed files with 5 additions and 1 deletions
@@ -231,7 +231,11 @@ hipError_t ihipModuleLaunchKernel(hipFunction_t f, uint32_t globalWorkSizeX,
        "%s", "Both, kernelParams and extra Params are provided, only one should be provided");
    return hipErrorInvalidValue;
  }
-
+  if (globalWorkSizeX == 0 || globalWorkSizeY == 0 || globalWorkSizeZ == 0 ||
+      blockDimX == 0 || blockDimY == 0 || blockDimZ == 0) {
+    return hipErrorInvalidValue;
+  }
+  
  hip::DeviceFunc* function = hip::DeviceFunc::asFunction(f);
  amd::Kernel* kernel = function->kernel();