Lines Matching defs:unrollFactor

320 /// associated 'forOpIV' by 'unrollFactor', calling 'ivRemapFn' to remap
324     Block *loopBodyBlock, Value forOpIV, uint64_t unrollFactor,
340   // Unroll the contents of 'forOp' (append unrollFactor - 1 additional copies).
343   for (unsigned i = 1; i < unrollFactor; i++) {
376 /// Unrolls 'forOp' by 'unrollFactor', returns the unrolled main loop and the
379     scf::ForOp forOp, uint64_t unrollFactor,
381   assert(unrollFactor > 0 && "expected positive unroll factor");
403     if (unrollFactor == 1) {
411         *constTripCount - (*constTripCount % unrollFactor);
413     int64_t stepUnrolledCst = stepCst * unrollFactor;
440         loc, boundsBuilder.getIntegerAttr(tripCount.getType(), unrollFactor));
443     // Compute tripCountEvenMultiple = tripCount - (tripCount % unrollFactor)
450     // Scale 'step' by 'unrollFactor'.
485       forOp.getBody(), forOp.getInductionVar(), unrollFactor,