SWDEV-354717 - Fix hipExtLaunchKernel as part of stream capture

Change-Id: Ic29d7403dfa38f01bec423aa1df7aa75c60a86e2


[ROCm/clr commit: 7f4c711ac1]

projects/clr/hipamd/src/hip_graph.cpp

@@ -274,10 +274,10 @@ hipError_t capturehipExtLaunchKernel(hipStream_t& stream, const void*& hostFunct
                                      hipEvent_t& startEvent, hipEvent_t& stopEvent, int& flags) {
   ClPrint(amd::LOG_INFO, amd::LOG_API,
           "[hipGraph] current capture node Ext kernel launch on stream : %p", stream);
-  return ihipExtLaunchKernel(stream,
-                             reinterpret_cast<hipFunction_t>(const_cast<void*>(hostFunction)),
-                             gridDim.x, gridDim.y, gridDim.z, blockDim.x, blockDim.y, blockDim.z,
-                             sharedMemBytes, args, nullptr, startEvent, stopEvent, flags);
+  return ihipExtLaunchKernel(
+      stream, reinterpret_cast<hipFunction_t>(const_cast<void*>(hostFunction)),
+      gridDim.x * blockDim.x, gridDim.y * blockDim.y, gridDim.z * blockDim.z, blockDim.x,
+      blockDim.y, blockDim.z, sharedMemBytes, args, nullptr, startEvent, stopEvent, flags);
 }
 
 hipError_t capturehipModuleLaunchKernel(hipStream_t& stream, hipFunction_t& f, uint32_t& gridDimX,