diff --git a/mlir/lib/Conversion/GPUCommon/GPUToLLVMConversion.cpp b/mlir/lib/Conversion/GPUCommon/GPUToLLVMConversion.cpp --- a/mlir/lib/Conversion/GPUCommon/GPUToLLVMConversion.cpp +++ b/mlir/lib/Conversion/GPUCommon/GPUToLLVMConversion.cpp @@ -752,10 +752,10 @@ auto kernelParams = generateParamsArray(launchOp, operands, rewriter); auto nullpointer = rewriter.create(loc, llvmPointerPointerType); launchKernelCallBuilder.create(loc, rewriter, - {function.getResult(0), launchOp.gridSizeX(), - launchOp.gridSizeY(), launchOp.gridSizeZ(), - launchOp.blockSizeX(), launchOp.blockSizeY(), - launchOp.blockSizeZ(), + {function.getResult(0), adaptor.gridSizeX(), + adaptor.gridSizeY(), adaptor.gridSizeZ(), + adaptor.blockSizeX(), adaptor.blockSizeY(), + adaptor.blockSizeZ(), /*sharedMemBytes=*/zero, stream, kernelParams, /*extra=*/nullpointer});