| /llvm-project-15.0.7/mlir/lib/Dialect/SCF/Transforms/ |
| H A D | ForToWhile.cpp | 48 auto whileOp = rewriter.create<WhileOp>(forOp.getLoc(), lcvTypes, initArgs, in matchAndRewrite() local
54 &whileOp.getBefore(), whileOp.getBefore().begin(), lcvTypes, lcvLocs); in matchAndRewrite()
55 rewriter.setInsertionPointToStart(&whileOp.getBefore().front()); in matchAndRewrite()
57 whileOp.getLoc(), arith::CmpIPredicate::slt, in matchAndRewrite()
59 rewriter.create<scf::ConditionOp>(whileOp.getLoc(), cmpOp.getResult(), in matchAndRewrite()
66 &whileOp.getAfter(), whileOp.getAfter().begin(), lcvTypes, lcvLocs); in matchAndRewrite()
71 whileOp.getLoc(), afterBlock->getArgument(0), forOp.getStep()); in matchAndRewrite()
94 arg.value().replaceAllUsesWith(whileOp.getResult(arg.index() + 1)); in matchAndRewrite()
|
| H A D | BufferizableOpInterfaceImpl.cpp | 575 auto whileOp = cast<scf::WhileOp>(op); in getAliasingOpResult() local
585 return {whileOp->getResult(idx)}; in getAliasingOpResult()
594 auto whileOp = cast<scf::WhileOp>(op); in bufferRelation() local
603 auto conditionOp = whileOp.getConditionOp(); in bufferRelation()
609 auto yieldOp = whileOp.getYieldOp(); in bufferRelation()
647 auto whileOp = cast<scf::WhileOp>(op); in resolveConflicts() local
649 auto yieldOp = whileOp.getYieldOp(); in resolveConflicts()
662 whileOp.getAfterArguments(), whileOp.getYieldOp().getResults(), state); in resolveConflicts()
714 auto whileOp = cast<scf::WhileOp>(op); in bufferize() local
818 auto whileOp = cast<scf::WhileOp>(op); in verifyAnalysis() local
[all …]
|
| /llvm-project-15.0.7/mlir/lib/Conversion/SCFToControlFlow/ |
| H A D | SCFToControlFlow.cpp | 274 LogicalResult matchAndRewrite(WhileOp whileOp,
285 LogicalResult matchAndRewrite(WhileOp whileOp,
526 Location loc = whileOp.getLoc(); in matchAndRewrite()
534 Block *after = &whileOp.getAfter().front(); in matchAndRewrite()
535 Block *afterLast = &whileOp.getAfter().back(); in matchAndRewrite()
536 Block *before = &whileOp.getBefore().front(); in matchAndRewrite()
570 return rewriter.notifyMatchFailure(whileOp, in matchAndRewrite()
576 return rewriter.notifyMatchFailure(whileOp, in matchAndRewrite()
582 return rewriter.notifyMatchFailure(whileOp, in matchAndRewrite()
593 Block *before = &whileOp.getBefore().front(); in matchAndRewrite()
[all …]
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Math/Transforms/ |
| H A D | ExpandPatterns.cpp | 74 auto whileOp = rewriter.create<scf::WhileOp>(loc, types, operands); in convertCtlzOp() local
76 rewriter.createBlock(&whileOp.getBefore(), {}, types, locations); in convertCtlzOp()
78 rewriter.createBlock(&whileOp.getAfter(), {}, types, locations); in convertCtlzOp()
82 rewriter.setInsertionPointToStart(&whileOp.getBefore().front()); in convertCtlzOp()
94 rewriter.setInsertionPointToStart(&whileOp.getAfter().front()); in convertCtlzOp()
109 rewriter.setInsertionPointAfter(whileOp); in convertCtlzOp()
110 rewriter.replaceOp(op, whileOp->getResult(1)); in convertCtlzOp()
|
| /llvm-project-15.0.7/flang/lib/Optimizer/Transforms/ |
| H A D | RewriteLoop.cpp | 210 matchAndRewrite(fir::IterWhileOp whileOp, in matchAndRewrite() argument
212 auto loc = whileOp.getLoc(); in matchAndRewrite()
226 auto *conditionBlock = &whileOp.getRegion().front(); in matchAndRewrite()
229 auto *lastBodyBlock = &whileOp.getRegion().back(); in matchAndRewrite()
239 auto step = whileOp.getStep(); in matchAndRewrite()
245 auto begin = whileOp.getFinalValue() in matchAndRewrite()
254 auto lowerBound = whileOp.getLowerBound(); in matchAndRewrite()
255 auto upperBound = whileOp.getUpperBound(); in matchAndRewrite()
262 auto iterOperands = whileOp.getIterOperands(); in matchAndRewrite()
291 auto args = whileOp.getFinalValue() in matchAndRewrite()
[all …]
|
| /llvm-project-15.0.7/mlir/lib/Conversion/SCFToSPIRV/ |
| H A D | SCFToSPIRV.cpp | 339 WhileOpConversion::matchAndRewrite(scf::WhileOp whileOp, OpAdaptor adaptor, in matchAndRewrite() argument
341 auto loc = whileOp.getLoc(); in matchAndRewrite()
347 Region &beforeRegion = whileOp.getBefore(); in matchAndRewrite()
348 Region &afterRegion = whileOp.getAfter(); in matchAndRewrite()
421 rewriter.replaceOp(whileOp, resultValues); in matchAndRewrite()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/SparseTensor/Transforms/ |
| H A D | Sparsification.cpp | 1301 builder.setInsertionPointToStart(&whileOp.getBefore().front()); in genWhile()
1324 builder.setInsertionPointToStart(&whileOp.getAfter().front()); in genWhile()
1325 return whileOp; in genWhile()
1431 builder.setInsertionPointToEnd(&whileOp.getAfter().front()); in genWhileInduction()
1451 codegen.pidxs[tensor][idx] = whileOp->getResult(o++); in genWhileInduction()
1456 updateReduc(merger, codegen, whileOp->getResult(o++)); in genWhileInduction()
1460 codegen.expCount = whileOp->getResult(o++); in genWhileInduction()
1465 codegen.loops[idx] = whileOp->getResult(o++); in genWhileInduction()
1469 builder.setInsertionPointAfter(whileOp); in genWhileInduction()
1597 if (auto whileOp = dyn_cast<scf::WhileOp>(loop)) { in endLoop() local
[all …]
|
| H A D | SparseTensorConversion.cpp | 495 auto whileOp = rewriter.create<scf::WhileOp>(loc, noTypes, noArgs); in genSparse2SparseReshape() local
496 Block *before = rewriter.createBlock(&whileOp.getBefore(), {}, noTypes); in genSparse2SparseReshape()
502 Block *after = rewriter.createBlock(&whileOp.getAfter(), {}, noTypes); in genSparse2SparseReshape()
508 rewriter.setInsertionPointAfter(whileOp); in genSparse2SparseReshape()
761 auto whileOp = rewriter.create<scf::WhileOp>(loc, noTypes, noArgs); in matchAndRewrite() local
762 Block *before = rewriter.createBlock(&whileOp.getBefore(), {}, noTypes); in matchAndRewrite()
766 Block *after = rewriter.createBlock(&whileOp.getAfter(), {}, noTypes); in matchAndRewrite()
770 rewriter.setInsertionPointAfter(whileOp); in matchAndRewrite()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Tosa/Transforms/ |
| H A D | TosaInferShapes.cpp | 76 WhileOp whileOp = dyn_cast<WhileOp>(op); in propagateShapesToTosaWhile() local
77 if (!whileOp) in propagateShapesToTosaWhile()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Async/Transforms/ |
| H A D | AsyncParallelFor.cpp | 510 scf::WhileOp whileOp = b.create<scf::WhileOp>(types, operands); in createAsyncDispatchFunction() local
511 Block *before = b.createBlock(&whileOp.getBefore(), {}, types, locations); in createAsyncDispatchFunction()
512 Block *after = b.createBlock(&whileOp.getAfter(), {}, types, locations); in createAsyncDispatchFunction()
559 b.setInsertionPointAfter(whileOp); in createAsyncDispatchFunction()
|