Home
last modified time | relevance | path

Searched refs:partialIteration (Results 1 – 3 of 3) sorted by relevance

/llvm-project-15.0.7/mlir/lib/Dialect/SCF/Transforms/
H A DLoopSpecialization.cpp138 partialIteration = cast<ForOp>(b.clone(*forOp.getOperation())); in peelForLoop()
139 partialIteration.getLowerBoundMutable().assign(splitBound); in peelForLoop()
140 forOp.replaceAllUsesWith(partialIteration->getResults()); in peelForLoop()
141 partialIteration.getInitArgsMutable().assign(forOp->getResults()); in peelForLoop()
152 ForOp partialIteration, in rewriteAffineOpAfterPeeling() argument
155 Value partialIv = partialIteration.getInductionVar(); in rewriteAffineOpAfterPeeling()
156 assert(forOp.getStep() == partialIteration.getStep() && in rewriteAffineOpAfterPeeling()
167 partialIteration.walk([&](OpTy affineOp) { in rewriteAffineOpAfterPeeling()
185 rewriter, forOp, partialIteration, previousUb); in peelAndCanonicalizeForLoop()
187 rewriter, forOp, partialIteration, previousUb); in peelAndCanonicalizeForLoop()
[all …]
/llvm-project-15.0.7/mlir/include/mlir/Dialect/SCF/Transforms/
H A DTransforms.h89 scf::ForOp &partialIteration);
/llvm-project-15.0.7/mlir/lib/Dialect/Linalg/Transforms/
H A DTransforms.cpp315 scf::ForOp partialIteration; in peelLoop() local
317 partialIteration))) in peelLoop()
318 return partialIteration->getResults(); in peelLoop()
319 assert(!partialIteration && "expected that loop was not peeled"); in peelLoop()