SWDEV-382359 - Workaround a non-uniform launch in MIOpen

Change-Id: Idee06494a67e641301890863b376d9ef5965837d
ROCm · Mar 28, 2023 · c1741e9 · c1741e9
1 parent bb2b6bb
commit c1741e9
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/src/hip_module.cpp b/src/hip_module.cpp
@@ -368,6 +368,12 @@ hipError_t ihipModuleLaunchKernel(hipFunction_t f, uint32_t globalWorkSizeX,
   if (status != hipSuccess) {
     return status;
   }
+
+  // Make sure the app doesn't launch a workgroup bigger than the global size
+  if (globalWorkSizeX < blockDimX) blockDimX = globalWorkSizeX;
+  if (globalWorkSizeY < blockDimY) blockDimY = globalWorkSizeY;
+  if (globalWorkSizeZ < blockDimZ) blockDimZ = globalWorkSizeZ;
+
   amd::Command* command = nullptr;
   amd::HostQueue* queue = hip::getQueue(hStream);
   status = ihipLaunchKernelCommand(command, f, globalWorkSizeX, globalWorkSizeY, globalWorkSizeZ,