Lines Matching defs:loopRange
193 static bool tileDividesIterationDomain(Range loopRange) {
194 std::optional<int64_t> offsetAsInt = getConstantIntValue(loopRange.offset);
197 std::optional<int64_t> sizeAsInt = getConstantIntValue(loopRange.size);
200 std::optional<int64_t> strideAsInt = getConstantIntValue(loopRange.stride);
206 /// Returns the bounded tile size given the current `offset`, `loopRange` and
209 Range loopRange, OpFoldResult offset,
216 Range{loopRange.offset, loopRange.size, tileSize}))
226 Value size = getValueOrCreateConstantIndexOp(b, loc, loopRange.size);
263 for (auto [nt, tileSize, loopRange] :
267 // `loopRange.offset/size`.
269 offsets.push_back(loopRange.offset);
270 sizes.push_back(loopRange.size);
277 ArrayRef<OpFoldResult>{loopRange.offset, iv, tileSize});
280 {loopRange.offset, nt, tileSize, loopRange.size});
286 {offset, loopRange.size});
302 if (!canOmitTileOffsetInBoundsCheck(tileSize, nt, loopRange.size)) {
314 for (auto [tileSize, loopRange] :
318 // `loopRange.offset/size`.
320 offsets.push_back(loopRange.offset);
321 sizes.push_back(loopRange.size);
329 getBoundedTileSize(rewriter, loc, loopRange, offset, tileSize);
342 for (auto [loopRange, tileSize] : llvm::zip_equal(loopRanges, tileSizes)) {
346 lbs.push_back(loopRange.offset);
347 ubs.push_back(loopRange.size);
2207 for (auto loopRange : domain) {
2209 getValueOrCreateConstantIndexOp(rewriter, loc, loopRange.offset);
2211 getValueOrCreateConstantIndexOp(rewriter, loc, loopRange.size);
2213 getValueOrCreateConstantIndexOp(rewriter, loc, loopRange.stride);