| /llvm-project/mlir/lib/Dialect/SparseTensor/Transforms/ |
| H A D | SparseIterationToScf.cpp | 130 Region &loopBody, SparseIterator *it, in matchAndRewrite()
262 [block](PatternRewriter &rewriter, Location loc, Region &loopBody,
269 Block *dstBlock = &loopBody.getBlocks().front();
100 Block *loopBody = op.getBody(); matchAndRewrite() local
142 Block *loopBody = op.getBody(); matchAndRewrite() local
|
| /llvm-project/mlir/test/Dialect/Bufferization/Transforms/OwnershipBasedBufferDeallocation/ |
| H A D | invalid-buffer-deallocation.mlir | 20 ^loopBody(%i, %buff : i32, memref<?xf32>),
23 ^loopBody(%val : i32, %buff2: memref<?xf32>):
48 cf.br ^loopBody(%const0, %arg2 : i32, memref<2xf32>)
50 ^loopBody(%val : i32, %buff2: memref<2xf32>):
59 ^loopBody(%i, %buff : i32, memref<2xf32>),
|
| /llvm-project/mlir/lib/Conversion/BufferizationToMemRef/ |
| H A D | BufferizationToMemRef.cpp | 62 auto loopBody = [&](OpBuilder &builder, Location loc, Value i, in matchAndRewrite() local
74 loopBody) in matchAndRewrite()
|
| /llvm-project/mlir/lib/Dialect/Affine/Transforms/ |
| H A D | AffineLoopInvariantCodeMotion.cpp | 205 auto *loopBody = forOp.getBody(); runOnAffineForOp() local
|
| /llvm-project/mlir/test/Analysis/ |
| H A D | test-dominance.mlir | 132 cf.cond_br %lessThan, ^loopBody, ^exit
133 ^loopBody:
456 cf.cond_br %lessThan, ^loopBody, ^exit
457 ^loopBody:
|
| H A D | test-liveness.mlir | 131 cf.cond_br %lessThan, ^loopBody(%i : i32), ^exit(%i : i32)
132 ^loopBody(%val : i32):
|
| /llvm-project/mlir/lib/Conversion/ControlFlowToSCF/ |
| H A D | ControlFlowToSCF.cpp | 86 Value condition, ValueRange loopVariablesNextIter, Region &&loopBody) { in createStructuredDoWhileLoopOp()
91 whileOp.getBefore().takeBody(loopBody); in createStructuredDoWhileLoopOp()
87 createStructuredDoWhileLoopOp(OpBuilder & builder,Operation * replacedOp,ValueRange loopVariablesInit,Value condition,ValueRange loopVariablesNextIter,Region && loopBody) createStructuredDoWhileLoopOp() argument
|
| /llvm-project/mlir/include/mlir/Conversion/ControlFlowToSCF/ |
| H A D | ControlFlowToSCF.h | 46 Region &&loopBody) override;
|
| /llvm-project/mlir/test/CAPI/ |
| H A D | ir.c | 49 void populateLoopBody(MlirContext ctx, MlirBlock loopBody, in populateLoopBody() argument
51 MlirValue iv = mlirBlockGetArgument(loopBody, 0); in populateLoopBody()
63 mlirBlockAppendOwnedOperation(loopBody, loadLHS); in populateLoopBody()
71 mlirBlockAppendOwnedOperation(loopBody, loadRHS); in populateLoopBody()
80 mlirBlockAppendOwnedOperation(loopBody, add); in populateLoopBody()
87 mlirBlockAppendOwnedOperation(loopBody, store); in populateLoopBody()
92 mlirBlockAppendOwnedOperation(loopBody, yield); in populateLoopBody()
154 MlirBlock loopBody = mlirBlockCreate(0, NULL, NULL); in makeAndDumpAdd() local
155 mlirBlockAddArgument(loopBody, indexType, location); in makeAndDumpAdd()
156 mlirRegionAppendOwnedBlock(loopBodyRegion, loopBody); in makeAndDumpAdd()
[all...] |
| /llvm-project/mlir/lib/Transforms/Utils/ |
| H A D | CFGToSCF.cpp | 862 Region loopBody; in transformCyclesToSCFLoops() local
864 loopBody.push_back(blocks.remove(loopHeader)); in transformCyclesToSCFLoops()
867 loopBody.push_back(blocks.remove(block)); in transformCyclesToSCFLoops()
869 loopBody.push_back(blocks.remove(latchBlock)); in transformCyclesToSCFLoops()
878 loopProperties->condition, iterationValues, std::move(loopBody)); in transformCyclesToSCFLoops()
|
| /llvm-project/mlir/include/mlir/Transforms/ |
| H A D | CFGToSCF.h | 71 Value condition, ValueRange loopValuesNextIter, Region &&loopBody) = 0;
|
| /llvm-project/mlir/lib/Dialect/SCF/Transforms/ |
| H A D | TileUsingInterface.cpp | 751 Block *loopBody = loopOp.getBody(); in tileReductionUsingScf()
754 loopBody, newLoopBody, in tileReductionUsingScf()
755 newLoopBody->getArguments().take_front(loopBody->getNumArguments())); in tileReductionUsingScf()
805 Block *loopBody = loopOp.getBody(); in tileReductionUsingScf()
808 loopBody, newLoopBody,
809 newLoopBody->getArguments().take_front(loopBody->getNumArguments()));
347 Block *loopBody = loopOp.getBody(); yieldTiledValuesAndReplaceLoop() local
401 Block *loopBody = loopOp.getBody(); yieldTiledValuesAndReplaceLoop() local
|
| H A D | BufferizableOpInterfaceImpl.cpp | 763 Block *loopBody = newForOp.getBody(); in bufferize()
767 rewriter.setInsertionPointToStart(loopBody); in bufferize()
774 rewriter.mergeBlocks(oldLoopBody, loopBody, iterArgs); in bufferize()
759 Block *loopBody = newForOp.getBody(); bufferize() local
|
| /llvm-project/mlir/lib/Conversion/VectorToGPU/ |
| H A D | VectorToGPU.cpp | 1164 Block &loopBody = *newForOp.getBody(); in convertForOp()
1168 valueMapping[loopBody.getArgument(mapping.first + in convertForOp()
1170 loopBody.getArgument(mapping.second + newForOp.getNumInductionVars()); in convertForOp()
1162 Block &loopBody = *newForOp.getBody(); convertForOp() local
|
| /llvm-project/mlir/test/Dialect/Bufferization/Transforms/ |
| H A D | buffer-deallocation.mlir | 1129 ^loopBody(%i, %buff : i32, memref<?xf32>),
1132 ^loopBody(%val : i32, %buff2: memref<?xf32>):
1157 cf.br ^loopBody(%const0, %arg2 : i32, memref<2xf32>)
1159 ^loopBody(%val : i32, %buff2: memref<2xf32>):
1168 ^loopBody(%i, %buff : i32, memref<2xf32>),
|
| /llvm-project/mlir/lib/Dialect/Affine/IR/ |
| H A D | AffineOps.cpp | 2637 auto loopBody = [&](OpBuilder &nestedBuilder, Location nestedLoc, Value iv, in buildAffineLoopNestImpl()
2650 auto loop = loopCreatorFn(builder, loc, lbs[i], ubs[i], steps[i], loopBody);
2631 auto loopBody = [&](OpBuilder &nestedBuilder, Location nestedLoc, Value iv, buildAffineLoopNestImpl() local
|
| /llvm-project/mlir/lib/Dialect/SCF/IR/ |
| H A D | SCF.cpp | 1652 Block *loopBody = forallOp.getBody(); in matchAndRewrite()
1672 rewriter.mergeBlocks(loopBody, newLoopBody, newBlockArgs); in matchAndRewrite()
1658 Block *loopBody = forallOp.getBody(); matchAndRewrite() local
|