SWDEV-542905 - Update the appropriate grid size in capture hipExtModuleLaunchKernel (#740)
Этот коммит содержится в:
коммит произвёл
GitHub
родитель
9cd8757717
Коммит
c43ee71e96
@@ -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