Lines Matching defs:unrollFactor

44 getCleanupLoopLowerBound(AffineForOp forOp, unsigned unrollFactor,
71     bumpExprs[i] = (tripCountExpr - tripCountExpr % unrollFactor) * step;
901                                                  uint64_t unrollFactor) {
904       *mayBeConstantTripCount < unrollFactor)
906   return loopUnrollByFactor(forOp, unrollFactor);
910 /// 'forOpIV' by 'unrollFactor', calling 'ivRemapFn' to remap 'forOpIV' for each
914     Block *loopBodyBlock, Value forOpIV, uint64_t unrollFactor,
930   // Unroll the contents of 'forOp' (append unrollFactor - 1 additional copies).
933   for (unsigned i = 1; i < unrollFactor; i++) {
973 /// count is not a multiple of `unrollFactor`.
975                                                   uint64_t unrollFactor) {
993   getCleanupLoopLowerBound(forOp, unrollFactor, cleanupMap, cleanupOperands);
1010     AffineForOp forOp, uint64_t unrollFactor,
1013   assert(unrollFactor > 0 && "unroll factor should be positive");
1016   if (unrollFactor == 1) {
1028   if (mayBeConstantTripCount && *mayBeConstantTripCount < unrollFactor) {
1037   // Generate the cleanup loop if trip count isn't a multiple of unrollFactor.
1038   if (getLargestDivisorOfTripCount(forOp) % unrollFactor != 0) {
1049     if (failed(generateCleanupLoopForUnroll(forOp, unrollFactor)))
1059   forOp.setStep(step * unrollFactor);
1061       forOp.getBody(), forOp.getInductionVar(), unrollFactor,