Lines Matching defs:origLoops

423 static void constructTiledLoopNest(MutableArrayRef<AffineForOp> origLoops,
459   moveLoopBody(origLoops.back(), innermostPointLoop);
661     MutableArrayRef<AffineForOp> origLoops,
663   assert(!origLoops.empty() && "expected atleast one loop in band");
664   assert(origLoops.size() == tileSizes.size() &&
667   OpBuilder b(origLoops[0].getOperation());
668   unsigned width = origLoops.size();
672     setInterTileBoundsParametric(b, origLoops[i], newLoops[i], tileSizes[i]);
677     setIntraTileBoundsParametric(b, origLoops[i], newLoops[i],
688 constructTiledIndexSetHyperRect(MutableArrayRef<AffineForOp> origLoops,
691   assert(!origLoops.empty());
692   assert(origLoops.size() == tileSizes.size());
694   OpBuilder b(origLoops[0].getOperation());
695   unsigned width = origLoops.size();
699     OperandRange newLbOperands = origLoops[i].getLowerBoundOperands();
700     OperandRange newUbOperands = origLoops[i].getUpperBoundOperands();
701     newLoops[i].setLowerBound(newLbOperands, origLoops[i].getLowerBoundMap());
702     newLoops[i].setUpperBound(newUbOperands, origLoops[i].getUpperBoundMap());
705     newLoops[i].setStep(tileSizes[i] * origLoops[i].getStepAsInt());
709     int64_t largestDiv = getLargestDivisorOfTripCount(origLoops[i]);
711         getConstantTripCount(origLoops[i]);
717     newLoops[width + i].setStep(origLoops[i].getStepAsInt());
724           *mayBeConstantCount * origLoops[i].getStepAsInt());
736       AffineBound ub = origLoops[i].getUpperBound();
754       boundExprs.push_back(dim + tileSizes[i] * origLoops[i].getStepAsInt());
765           1, 0, dim + tileSizes[i] * origLoops[i].getStepAsInt());
781   MutableArrayRef<AffineForOp> origLoops = input;
782   AffineForOp rootAffineForOp = origLoops[0];
790   constructTiledLoopNest(origLoops, rootAffineForOp, width, tiledLoops);
796   constructTiledIndexSetHyperRect(origLoops, tiledLoops, tileSizes);
823   MutableArrayRef<AffineForOp> origLoops = input;
824   AffineForOp rootAffineForOp = origLoops[0];
830   constructTiledLoopNest(origLoops, rootAffineForOp, width, tiledLoops);
836   constructParametricallyTiledIndexSetHyperRect(origLoops, tiledLoops,