Lines Matching defs:launchOp

124     gpu::LaunchOp launchOp,
127   Region &launchOpBody = launchOp.getBody();
152                                  launchOp.getBody());
178 static gpu::GPUFuncOp outlineKernelFuncImpl(gpu::LaunchOp launchOp,
181   Location loc = launchOp.getLoc();
184   OpBuilder builder(launchOp.getContext());
185   Region &launchOpBody = launchOp.getBody();
198       FunctionType::get(launchOp.getContext(), kernelOperandTypes, {});
201       TypeRange(ValueRange(launchOp.getWorkgroupAttributions())),
202       TypeRange(ValueRange(launchOp.getPrivateAttributions())));
210           maybeConstantDimsAttr(launchOp.getBlockSizeOperandValues()))
213           maybeConstantDimsAttr(launchOp.getGridSizeOperandValues()))
223                            launchOp.hasClusterSize());
227        llvm::zip(launchOp.getWorkgroupAttributions(),
231        llvm::zip(launchOp.getPrivateAttributions(),
265 gpu::GPUFuncOp mlir::outlineKernelFunc(gpu::LaunchOp launchOp,
271   auto funcOp = outlineKernelFuncImpl(launchOp, kernelFnName, operandSet);
282 static void convertToLaunchFuncOp(gpu::LaunchOp launchOp,
285   OpBuilder builder(launchOp);
288   Value asyncToken = launchOp.getAsyncToken();
290       launchOp.getClusterSizeOperandValues();
292       launchOp.getLoc(), kernelFunc, launchOp.getGridSizeOperandValues(),
293       launchOp.getBlockSizeOperandValues(),
294       launchOp.getDynamicSharedMemorySize(), operands,
296       launchOp.getAsyncDependencies(), clusterSize);
297   launchOp.replaceAllUsesWith(launchFunc);
298   launchOp.erase();