Searched refs:afterBlock (Results 1 – 4 of 4) sorted by relevance
| /llvm-project-15.0.7/mlir/lib/Dialect/SCF/Transforms/ |
| H A D | ForToWhile.cpp | 65 auto *afterBlock = rewriter.createBlock( in matchAndRewrite() local
69 rewriter.setInsertionPointToEnd(afterBlock); in matchAndRewrite()
71 whileOp.getLoc(), afterBlock->getArgument(0), forOp.getStep()); in matchAndRewrite()
76 barg.value().replaceAllUsesWith(afterBlock->getArgument(barg.index())); in matchAndRewrite()
80 arg.moveBefore(afterBlock, afterBlock->end()); in matchAndRewrite()
83 for (auto yieldOp : afterBlock->getOps<scf::YieldOp>()) { in matchAndRewrite()
|
| /llvm-project-15.0.7/mlir/lib/Conversion/SCFToSPIRV/ |
| H A D | SCFToSPIRV.cpp | 352 Block &afterBlock = afterRegion.front(); in matchAndRewrite() local
364 auto yield = cast<scf::YieldOp>(afterBlock.getTerminator()); in matchAndRewrite()
415 cond, conditionVal, &afterBlock, condArgs, &mergeBlock, llvm::None); in matchAndRewrite()
418 rewriter.setInsertionPointToEnd(&afterBlock); in matchAndRewrite()
|
| /llvm-project-15.0.7/mlir/lib/Conversion/SCFToControlFlow/ |
| H A D | SCFToControlFlow.cpp | 574 Block &afterBlock = whileOp.getAfter().front(); in matchAndRewrite() local
575 if (!llvm::hasSingleElement(afterBlock)) in matchAndRewrite()
580 auto yield = dyn_cast<scf::YieldOp>(&afterBlock.front()); in matchAndRewrite()
581 if (!yield || yield.getResults() != afterBlock.getArguments()) in matchAndRewrite()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/SCF/IR/ |
| H A D | SCF.cpp | 2815 Block &afterBlock = op.getAfter().front(); in matchAndRewrite() local
2819 Operation *yieldOp = afterBlock.getTerminator(); in matchAndRewrite()
2840 if (yieldOpBlockArg && yieldOpBlockArg.getOwner() == &afterBlock) { in matchAndRewrite()
2873 if (yieldOpBlockArg && yieldOpBlockArg.getOwner() == &afterBlock) { in matchAndRewrite()
3022 Block &afterBlock = op.getAfter().front(); in matchAndRewrite() local
3027 SmallVector<Value> newAfterBlockArgs(afterBlock.getNumArguments()); in matchAndRewrite()
3028 SmallVector<Value> newWhileResults(afterBlock.getNumArguments()); in matchAndRewrite()
3029 for (unsigned i = 0, j = 0, n = afterBlock.getNumArguments(); i < n; i++) { in matchAndRewrite()
3045 rewriter.mergeBlocks(&afterBlock, &newAfterBlock, newAfterBlockArgs); in matchAndRewrite()
3139 Block &afterBlock = op.getAfter().front(); in matchAndRewrite() local
[all …]
|