Lines Matching refs:unrollFactor
58 static void getCleanupLoopLowerBound(AffineForOp forOp, unsigned unrollFactor, in getCleanupLoopLowerBound() argument
93 bumpExprs[i] = (tripCountExpr - tripCountExpr % unrollFactor) * step; in getCleanupLoopLowerBound()
1045 uint64_t unrollFactor) { in loopUnrollUpToFactor() argument
1048 mayBeConstantTripCount.getValue() < unrollFactor) in loopUnrollUpToFactor()
1050 return loopUnrollByFactor(forOp, unrollFactor); in loopUnrollUpToFactor()
1057 generateUnrolledLoop(Block *loopBodyBlock, Value forOpIV, uint64_t unrollFactor, in generateUnrolledLoop() argument
1071 for (unsigned i = 1; i < unrollFactor; i++) { in generateUnrolledLoop()
1100 uint64_t unrollFactor) { in loopUnrollByFactor() argument
1101 assert(unrollFactor > 0 && "unroll factor should be positive"); in loopUnrollByFactor()
1103 if (unrollFactor == 1) in loopUnrollByFactor()
1122 mayBeConstantTripCount.getValue() < unrollFactor) in loopUnrollByFactor()
1126 if (getLargestDivisorOfTripCount(forOp) % unrollFactor != 0) { in loopUnrollByFactor()
1131 getCleanupLoopLowerBound(forOp, unrollFactor, cleanupMap, cleanupOperands); in loopUnrollByFactor()
1146 forOp.setStep(step * unrollFactor); in loopUnrollByFactor()
1147 generateUnrolledLoop(forOp.getBody(), forOp.getInductionVar(), unrollFactor, in loopUnrollByFactor()
1164 uint64_t unrollFactor) { in loopUnrollByFactor() argument
1165 assert(unrollFactor > 0 && "expected positive unroll factor"); in loopUnrollByFactor()
1166 if (unrollFactor == 1) in loopUnrollByFactor()
1193 int64_t tripCountEvenMultiple = tripCount - (tripCount % unrollFactor); in loopUnrollByFactor()
1196 int64_t stepUnrolledCst = stepCst * unrollFactor; in loopUnrollByFactor()
1220 boundsBuilder.create<ConstantIndexOp>(loc, unrollFactor); in loopUnrollByFactor()
1261 forOp.getBody(), forOp.getInductionVar(), unrollFactor, in loopUnrollByFactor()