Lines Matching refs:launchOp
111 gpu::LaunchOp launchOp, in sinkOperationsIntoLaunchOp() argument
114 Region &launchOpBody = launchOp.body(); in sinkOperationsIntoLaunchOp()
139 launchOp.body()); in sinkOperationsIntoLaunchOp()
146 static gpu::GPUFuncOp outlineKernelFuncImpl(gpu::LaunchOp launchOp, in outlineKernelFuncImpl() argument
149 Location loc = launchOp.getLoc(); in outlineKernelFuncImpl()
152 OpBuilder builder(launchOp.getContext()); in outlineKernelFuncImpl()
153 Region &launchOpBody = launchOp.body(); in outlineKernelFuncImpl()
166 FunctionType::get(launchOp.getContext(), kernelOperandTypes, {}); in outlineKernelFuncImpl()
205 gpu::GPUFuncOp mlir::outlineKernelFunc(gpu::LaunchOp launchOp, in outlineKernelFunc() argument
211 auto funcOp = outlineKernelFuncImpl(launchOp, kernelFnName, operandSet); in outlineKernelFunc()
222 static void convertToLaunchFuncOp(gpu::LaunchOp launchOp, in convertToLaunchFuncOp() argument
225 OpBuilder builder(launchOp); in convertToLaunchFuncOp()
228 Value asyncToken = launchOp.asyncToken(); in convertToLaunchFuncOp()
230 launchOp.getLoc(), kernelFunc, launchOp.getGridSizeOperandValues(), in convertToLaunchFuncOp()
231 launchOp.getBlockSizeOperandValues(), launchOp.dynamicSharedMemorySize(), in convertToLaunchFuncOp()
233 launchOp.asyncDependencies()); in convertToLaunchFuncOp()
234 launchOp.replaceAllUsesWith(launchFunc); in convertToLaunchFuncOp()
235 launchOp.erase(); in convertToLaunchFuncOp()