Lines Matching defs:unrollJamFactor

517                                           uint64_t unrollJamFactor) {
518   assert(unrollJamFactor > 0 && "unroll jam factor should be positive");
520   if (unrollJamFactor == 1)
544   if (unrollJamFactor > *tripCount) {
547     unrollJamFactor = *tripCount;
548   } else if (*tripCount % unrollJamFactor != 0) {
568   // iteration. There are (`unrollJamFactor` - 1) iterations.
569   SmallVector<IRMapping> operandMaps(unrollJamFactor - 1);
572   // `unrollJamFactor` copies of its iterOperands, iter_args and yield
584     for (unsigned i = unrollJamFactor - 1; i >= 1; --i) {
605     unsigned oldNumResults = newResults.size() / unrollJamFactor;
608     for (unsigned i = unrollJamFactor - 1; i >= 1; --i) {
610         // `newForOp` has `unrollJamFactor` - 1 new sets of iterArgs and
627           forOp.getLoc(), rewriter.getIndexAttr(unrollJamFactor)));
631   // Unroll and jam (appends unrollJamFactor - 1 additional copies).
632   for (unsigned i = unrollJamFactor - 1; i >= 1; --i) {
641         // iv' = iv + i * step, i = 1 to unrollJamFactor-1.
655           newForOp.getNumRegionIterArgs() / unrollJamFactor;
658       unsigned oldNumYieldOperands = yieldOp.getNumOperands() / unrollJamFactor;