From 459e9539b240e1d117f0e33d01945f01529ccf42 Mon Sep 17 00:00:00 2001 From: Anusha Godavarthy Surya Date: Thu, 1 Sep 2022 19:26:14 +0000 Subject: [PATCH] SWDEV-354717 - Fix hipExtLaunchKernel as part of stream capture Change-Id: Ic29d7403dfa38f01bec423aa1df7aa75c60a86e2 [ROCm/clr commit: 7f4c711ac11cd7a227598b722322f7f8baf9c82d] --- projects/clr/hipamd/src/hip_graph.cpp | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/projects/clr/hipamd/src/hip_graph.cpp b/projects/clr/hipamd/src/hip_graph.cpp index edf6ee97c8..f59bdd97e5 100644 --- a/projects/clr/hipamd/src/hip_graph.cpp +++ b/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(const_cast(hostFunction)), - gridDim.x, gridDim.y, gridDim.z, blockDim.x, blockDim.y, blockDim.z, - sharedMemBytes, args, nullptr, startEvent, stopEvent, flags); + return ihipExtLaunchKernel( + stream, reinterpret_cast(const_cast(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,