diff --git a/projects/clr/rocclr/device/rocm/rocblit.cpp b/projects/clr/rocclr/device/rocm/rocblit.cpp index 5862f701f1..1fcd23c6fe 100644 --- a/projects/clr/rocclr/device/rocm/rocblit.cpp +++ b/projects/clr/rocclr/device/rocm/rocblit.cpp @@ -2540,7 +2540,7 @@ bool KernelBlitManager::batchMemOps(const void* paramArray, size_t paramSize, memcpy(constBuf, paramArray, (count * paramSize)); setArgument(kernels_[blitType], 0, sizeof(cl_mem), constBuf, 0, nullptr, kDirectVa); - setArgument(kernels_[blitType], 1, sizeof(cl_mem), &count); + setArgument(kernels_[blitType], 1, sizeof(uint32_t), &count); // Create ND range object for the kernel's execution amd::NDRangeContainer ndrange(dim, globalWorkOffset, globalWorkSize, localWorkSize);