SWDEV-352951 - Add capture support for hipModuleLaunchKernel

Change-Id: Iced0846c9f753ecbb87e3f7aec2bfd72137d8c00
此提交包含在:
Anusha Godavarthy Surya
2022-08-22 03:22:44 -07:00
提交者 Anusha Godavarthy Surya
父節點 c833a19284
當前提交 c72942964d
共有 3 個檔案被更改,包括 40 行新增 和 0 行删除
+4
查看文件
@@ -396,6 +396,10 @@ hipError_t hipModuleLaunchKernel(hipFunction_t f, uint32_t gridDimX, uint32_t gr
void** kernelParams, void** extra) {
HIP_INIT_API(hipModuleLaunchKernel, f, gridDimX, gridDimY, gridDimZ, blockDimX, blockDimY,
blockDimZ, sharedMemBytes, hStream, kernelParams, extra);
STREAM_CAPTURE(hipModuleLaunchKernel, hStream, f, gridDimX, gridDimY, gridDimZ, blockDimX,
blockDimY, blockDimZ, sharedMemBytes, kernelParams, extra);
size_t globalWorkSizeX = static_cast<size_t>(gridDimX) * blockDimX;
size_t globalWorkSizeY = static_cast<size_t>(gridDimY) * blockDimY;
size_t globalWorkSizeZ = static_cast<size_t>(gridDimZ) * blockDimZ;