Home
last modified time | relevance | path

Searched refs:newLoop (Results 1 – 5 of 5) sorted by relevance

/llvm-project-15.0.7/mlir/lib/Dialect/Affine/Utils/
H A DLoopFusionUtils.cpp383 AffineForOp newLoop = replaceForOpWithNewYields( in promoteSingleIterReductionLoop() local
401 newLoop.getResult(i + parentOp->getNumResults())); in promoteSingleIterReductionLoop()
408 op->moveAfter(newLoop); in promoteSingleIterReductionLoop()
412 iv.replaceAllUsesWith(newLoop.getInductionVar()); in promoteSingleIterReductionLoop()
415 for (auto it : llvm::zip(forOpIterArgs, newLoop.getRegionIterArgs().take_back( in promoteSingleIterReductionLoop()
H A DLoopUtils.cpp645 AffineForOp newLoop, Value tileSize) { in setInterTileBoundsParametric() argument
650 newLoop.setLowerBound(newLbOperands, origLoop.getLowerBoundMap()); in setInterTileBoundsParametric()
733 newLoop.setUpperBound(ubOperands, ubMap); in setInterTileBoundsParametric()
736 newLoop.setStep(origLoop.getStep()); in setInterTileBoundsParametric()
/llvm-project-15.0.7/mlir/lib/Conversion/VectorToGPU/
H A DVectorToGPU.cpp737 scf::ForOp newLoop = in replaceForOpWithNewSignature() local
740 newLoop.getBody()->erase(); in replaceForOpWithNewSignature()
741 newLoop.getLoopBody().getBlocks().splice( in replaceForOpWithNewSignature()
742 newLoop.getLoopBody().getBlocks().begin(), in replaceForOpWithNewSignature()
745 newLoop.getBody()->addArgument(operand.getType(), operand.getLoc()); in replaceForOpWithNewSignature()
747 for (auto it : llvm::zip(loop.getResults(), newLoop.getResults().take_front( in replaceForOpWithNewSignature()
751 return newLoop; in replaceForOpWithNewSignature()
/llvm-project-15.0.7/mlir/lib/Dialect/SCF/Utils/
H A DUtils.cpp49 scf::ForOp newLoop = builder.create<scf::ForOp>( in replaceLoopWithNewYields() local
54 Block *newLoopBody = newLoop.getBody(); in replaceLoopWithNewYields()
87 return newLoop->isProperAncestor(user); in replaceLoopWithNewYields()
94 newLoop.getResults().take_front(loop.getNumResults())); in replaceLoopWithNewYields()
102 return newLoop; in replaceLoopWithNewYields()
/llvm-project-15.0.7/mlir/lib/Dialect/Affine/IR/
H A DAffineOps.cpp2252 AffineForOp newLoop = in replaceForOpWithNewYields() local
2256 newLoop.getLoopBody().takeBody(loop.getLoopBody()); in replaceForOpWithNewYields()
2258 newLoop.getLoopBody().addArgument(val.getType(), val.getLoc()); in replaceForOpWithNewYields()
2262 auto yield = cast<AffineYieldOp>(newLoop.getBody()->getTerminator()); in replaceForOpWithNewYields()
2270 for (auto it : llvm::zip(loop.getResults(), newLoop.getResults().take_front( in replaceForOpWithNewYields()
2275 return newLoop; in replaceForOpWithNewYields()