Adding changes for hipExtLaunchKernel for rocCLR

Change-Id: Iba52bc3bde7c37f3fb375a55ba0947e87b3cdc9b


[ROCm/hip commit: 2d517fdcc6]

projects/hip/src/hip_module.cpp

@@ -1709,12 +1709,33 @@ hipError_t hipLaunchKernel(
     const void* func_addr, dim3 numBlocks, dim3 dimBlocks, void** args,
     size_t sharedMemBytes, hipStream_t stream)
 {
-   HIP_INIT_API(hipLaunchKernel,func_addr,numBlocks,dimBlocks,args,sharedMemBytes,stream);
+    HIP_INIT_API(hipLaunchKernel,func_addr,numBlocks,dimBlocks,args,sharedMemBytes,stream);
 
-   hipFunction_t kd = hip_impl::get_program_state().kernel_descriptor((std::uintptr_t)func_addr,
+    hipFunction_t kd = hip_impl::get_program_state().kernel_descriptor((std::uintptr_t)func_addr,
                                                            hip_impl::target_agent(stream));
 
-   return hipModuleLaunchKernel(kd, numBlocks.x, numBlocks.y, numBlocks.z,
+    return hipModuleLaunchKernel(kd, numBlocks.x, numBlocks.y, numBlocks.z,
                           dimBlocks.x, dimBlocks.y, dimBlocks.z, sharedMemBytes,
                           stream, args, nullptr);
 }
+
+hipError_t hipExtLaunchKernel(const void* function, dim3 numBlocks, dim3 dimBlocks, void** args,
+                              size_t sharedMemBytes, hipStream_t stream, hipEvent_t startEvent,
+                              hipEvent_t stopEvent, int flags) {
+    HIP_INIT_API(hipExtLaunchKernel,function,numBlocks,dimBlocks,args,sharedMemBytes,stream,startEvent,stopEvent,flags);
+
+    hipFunction_t kd = hip_impl::get_program_state().kernel_descriptor((std::uintptr_t)function,
+                                                           hip_impl::target_agent(stream));
+
+    uint32_t globalWorkSizeX = numBlocks.x * dimBlocks.x;
+    uint32_t globalWorkSizeY = numBlocks.y * dimBlocks.y;
+    uint32_t globalWorkSizeZ = numBlocks.z * dimBlocks.z;
+    if (globalWorkSizeX > UINT32_MAX || globalWorkSizeY > UINT32_MAX ||
+        globalWorkSizeZ > UINT32_MAX) {
+        return hipErrorInvalidConfiguration;
+    }
+
+    return ihipLogStatus(ihipModuleLaunchKernel(
+        tls, kd, globalWorkSizeX, globalWorkSizeY, globalWorkSizeZ, dimBlocks.x, dimBlocks.y,
+        dimBlocks.z, sharedMemBytes, stream, args, nullptr, startEvent, stopEvent, flags));
+}