diff --git a/sycl/plugins/rocm/pi_rocm.cpp b/sycl/plugins/rocm/pi_rocm.cpp index 687bd2ed935d8..512bacd393bc5 100644 --- a/sycl/plugins/rocm/pi_rocm.cpp +++ b/sycl/plugins/rocm/pi_rocm.cpp @@ -2408,7 +2408,8 @@ pi_result rocm_piEnqueueKernelLaunch( } retError = PI_CHECK_ERROR(hipModuleLaunchKernel( - hipFunc, blocksPerGrid[0], 1, 1, threadsPerBlock[0], 1, 1, + hipFunc, blocksPerGrid[0], blocksPerGrid[1], blocksPerGrid[2], + threadsPerBlock[0], threadsPerBlock[1], threadsPerBlock[2], kernel->get_local_size(), hipStream, argIndices.data(), nullptr)); kernel->clear_local_size();