Lines Matching refs:unrollJamFactor
1150 uint64_t unrollJamFactor) { in loopUnrollJamUpToFactor() argument
1153 mayBeConstantTripCount.value() < unrollJamFactor) in loopUnrollJamUpToFactor()
1155 return loopUnrollJamByFactor(forOp, unrollJamFactor); in loopUnrollJamUpToFactor()
1201 uint64_t unrollJamFactor) { in loopUnrollJamByFactor() argument
1202 assert(unrollJamFactor > 0 && "unroll jam factor should be positive"); in loopUnrollJamByFactor()
1205 if (unrollJamFactor == 1) { in loopUnrollJamByFactor()
1217 if (mayBeConstantTripCount && *mayBeConstantTripCount < unrollJamFactor) { in loopUnrollJamByFactor()
1246 if (getLargestDivisorOfTripCount(forOp) % unrollJamFactor != 0) { in loopUnrollJamByFactor()
1254 if (failed(generateCleanupLoopForUnroll(forOp, unrollJamFactor))) in loopUnrollJamByFactor()
1261 SmallVector<BlockAndValueMapping, 4> operandMaps(unrollJamFactor - 1); in loopUnrollJamByFactor()
1276 for (unsigned i = unrollJamFactor - 1; i >= 1; --i) { in loopUnrollJamByFactor()
1295 unsigned oldNumResults = newResults.size() / unrollJamFactor; in loopUnrollJamByFactor()
1298 for (unsigned i = unrollJamFactor - 1; i >= 1; --i) { in loopUnrollJamByFactor()
1313 forOp.setStep(step * unrollJamFactor); in loopUnrollJamByFactor()
1317 for (unsigned i = unrollJamFactor - 1; i >= 1; --i) { in loopUnrollJamByFactor()
1340 newForOp.getNumIterOperands() / unrollJamFactor; in loopUnrollJamByFactor()
1343 unsigned oldNumYieldOperands = yieldOp.getNumOperands() / unrollJamFactor; in loopUnrollJamByFactor()
1366 unsigned oldNumResults = forOp.getNumResults() / unrollJamFactor; in loopUnrollJamByFactor()
1372 for (unsigned i = unrollJamFactor - 1; i >= 1; --i) { in loopUnrollJamByFactor()