Lines Matching refs:forOp
40 static Value getSupportedReduction(AffineForOp forOp, unsigned pos, in getSupportedReduction() argument
44 matchReduction(forOp.getRegionIterArgs(), pos, combinerOps); in getSupportedReduction()
81 AffineForOp forOp, SmallVectorImpl<LoopReduction> &supportedReductions) { in getSupportedReductions() argument
82 unsigned numIterArgs = forOp.getNumIterOperands(); in getSupportedReductions()
88 if (Value value = getSupportedReduction(forOp, i, kind)) in getSupportedReductions()
96 bool mlir::isLoopParallel(AffineForOp forOp, in isLoopParallel() argument
98 unsigned numIterArgs = forOp.getNumIterOperands(); in isLoopParallel()
107 getSupportedReductions(forOp, *parallelReductions); in isLoopParallel()
115 return isLoopMemoryParallel(forOp); in isLoopParallel()
134 bool mlir::isLoopMemoryParallel(AffineForOp forOp) { in isLoopMemoryParallel() argument
136 if (llvm::any_of(forOp.getResultTypes(), in isLoopMemoryParallel()
142 auto walkResult = forOp.walk([&](Operation *op) -> WalkResult { in isLoopMemoryParallel()
145 if (!isLocallyDefined(readOp.getMemRef(), forOp)) in isLoopMemoryParallel()
149 if (!isLocallyDefined(writeOp.getMemRef(), forOp)) in isLoopMemoryParallel()
167 unsigned depth = getNestingDepth(forOp) + 1; in isLoopMemoryParallel()
248 if (AffineForOp forOp = dyn_cast<AffineForOp>(op)) in getIndexSet() local
249 forOps.push_back(forOp); in getIndexSet()
256 if (AffineForOp forOp = dyn_cast<AffineForOp>(op)) { in getIndexSet() local
257 if (failed(domain->addAffineForOpDomain(forOp))) in getIndexSet()
331 AffineForOp forOp = getForInductionVarOwner(commonForIV); in getCommonBlock() local
332 assert(forOp && "commonForValue was not an induction variable"); in getCommonBlock()
336 getChainOfAncestorBlocks(srcAccess, srcAncestorBlocks, forOp.getBody()); in getCommonBlock()
337 getChainOfAncestorBlocks(dstAccess, dstAncestorBlocks, forOp.getBody()); in getCommonBlock()
339 Block *commonBlock = forOp.getBody(); in getCommonBlock()
665 AffineForOp forOp, unsigned maxLoopDepth, in getDependenceComponents() argument
669 forOp->walk([&](Operation *op) { in getDependenceComponents()