Searched refs:outlinedFunc (Results 1 – 2 of 2) sorted by relevance
| /external/llvm-project/mlir/lib/Dialect/GPU/Transforms/ |
| D | KernelOutlining.cpp | 159 auto outlinedFunc = builder.create<gpu::GPUFuncOp>(loc, kernelFnName, type); in outlineKernelFuncImpl() local
160 outlinedFunc.setAttr(gpu::GPUDialect::getKernelFuncAttrName(), in outlineKernelFuncImpl()
166 Region &outlinedFuncBody = outlinedFunc.body(); in outlineKernelFuncImpl()
189 outlinedFunc.walk([](gpu::TerminatorOp op) { in outlineKernelFuncImpl()
194 return outlinedFunc; in outlineKernelFuncImpl()
251 gpu::GPUFuncOp outlinedFunc = in runOnOperation() local
257 auto kernelModule = createKernelModule(outlinedFunc, symbolTable); in runOnOperation()
261 convertToLaunchFuncOp(op, outlinedFunc, operands.getArrayRef()); in runOnOperation()
|
| /external/llvm-project/mlir/lib/Dialect/SCF/Transforms/ |
| D | Utils.cpp | 100 auto outlinedFunc = b.create<FuncOp>(loc, funcName, type); in outlineIfOp() local
101 b.setInsertionPointToStart(outlinedFunc.addEntryBlock()); in outlineIfOp()
103 for (auto it : llvm::zip(values, outlinedFunc.getArguments())) in outlineIfOp()
116 Operation *call = b.create<CallOp>(loc, outlinedFunc, values); in outlineIfOp()
118 return outlinedFunc; in outlineIfOp()
|