SWDEV-542905 - Update the appropriate grid size in capture hipExtModuleLaunchKernel (#740)
[ROCm/clr commit: c43ee71e96]
This commit is contained in:
کامیت شده توسط
GitHub
والد
005672c314
کامیت
a3e576a317
@@ -296,7 +296,8 @@ hipError_t capturehipExtModuleLaunchKernel(hipStream_t& stream, hipFunction_t& f
|
||||
hipEvent_t& stopEvent, uint32_t& flags) {
|
||||
ClPrint(amd::LOG_INFO, amd::LOG_API,
|
||||
"[hipGraph] Current capture node ExtModuleLaunchKernel on stream : %p", stream);
|
||||
return ihipExtLaunchKernel(stream, f, globalWorkSizeX, globalWorkSizeY, globalWorkSizeZ,
|
||||
return ihipExtLaunchKernel(stream, f, globalWorkSizeX / localWorkSizeX,
|
||||
globalWorkSizeY / localWorkSizeY, globalWorkSizeZ / localWorkSizeZ,
|
||||
localWorkSizeX, localWorkSizeY, localWorkSizeZ, sharedMemBytes,
|
||||
kernelParams, extra, startEvent, stopEvent, flags);
|
||||
}
|
||||
|
||||
مرجع در شماره جدید
Block a user