Home
last modified time | relevance | path

Searched refs:lbs (Results 1 – 22 of 22) sorted by relevance

/llvm-project-15.0.7/mlir/lib/Dialect/GPU/Transforms/
H A DMemoryPromotion.cpp36 SmallVector<Value, 4> lbs, ubs, steps; in insertCopyLoops() local
44 lbs.resize(extraLoops, zero); in insertCopyLoops()
50 lbs.append(rank, zero); in insertCopyLoops()
51 ubs.reserve(lbs.size()); in insertCopyLoops()
52 steps.reserve(lbs.size()); in insertCopyLoops()
68 SmallVector<Value, 8> ivs(lbs.size()); in insertCopyLoops()
70 b, b.getLoc(), lbs, ubs, steps, in insertCopyLoops()
/llvm-project-15.0.7/mlir/lib/Dialect/Linalg/Utils/
H A DUtils.cpp136 lbs.emplace_back(range.offset); in unpackRanges()
526 SmallVector<Value, 4> lbs, ubs, steps; in doit() local
527 unpackRanges(loopRanges, lbs, ubs, steps); in doit()
569 SmallVector<Value, 4> lbs, ubs, steps; in doit() local
570 unpackRanges(loopRanges, lbs, ubs, steps); in doit()
616 assert(lbs.size() == ubs.size());
617 assert(lbs.size() == steps.size());
618 assert(lbs.size() == iteratorTypes.size());
697 Value cond = ab.slt(lbs[0], ubs[0]);
700 ivStorage.append(lbs.begin(), std::next(lbs.begin(), numProcessed));
[all …]
/llvm-project-15.0.7/mlir/lib/Dialect/Affine/Analysis/
H A DUtils.cpp120 lbs.clear(); in clearBounds()
132 for (auto &en : llvm::enumerate(lbs)) { in dump()
154 assert(lbs.size() == ubs.size() && !lbs.empty() && !ivs.empty() && in isSliceMaximalFastCheck()
157 for (unsigned i = 0, end = lbs.size(); i < end; ++i) { in isSliceMaximalFastCheck()
158 AffineMap lbMap = lbs[i]; in isSliceMaximalFastCheck()
394 if (lbs) { in getConstantBoundingSizeAndShape()
395 lbs->push_back(lb); in getConstantBoundingSizeAndShape()
931 sliceUnion->lbs.resize(numSliceLoopIVs, AffineMap()); in computeSliceUnion()
1001 AffineMap lbMap = slice.lbs[i]; in buildSliceTripCountMap()
1140 sliceState->lbs[j] = AffineMap(); in getComputationSliceState()
[all …]
H A DAffineStructures.cpp198 FlatAffineValueConstraints::getHyperrectangular(ValueRange ivs, ValueRange lbs, in getHyperrectangular() argument
202 assert(nIvs == lbs.size() && "expected as many lower bounds as ivs"); in getHyperrectangular()
209 unsigned lbsStart = res.appendDimVar(lbs); in getHyperrectangular()
/llvm-project-15.0.7/mlir/include/mlir/Dialect/SCF/IR/
H A DSCF.h89 OpBuilder &builder, Location loc, ValueRange lbs, ValueRange ubs,
98 LoopNest buildLoopNest(OpBuilder &builder, Location loc, ValueRange lbs,
/llvm-project-15.0.7/mlir/test/lib/Dialect/Affine/
H A DTestLoopFusion.cpp96 assert(sliceUnion.lbs.size() == sliceUnion.ubs.size()); in getSliceStr()
98 for (unsigned k = 0, e = sliceUnion.lbs.size(); k < e; ++k) { in getSliceStr()
100 sliceUnion.lbs[k].print(os); in getSliceStr()
/llvm-project-15.0.7/mlir/include/mlir/Dialect/Affine/Analysis/
H A DUtils.h80 SmallVector<AffineMap, 4> lbs; member
303 std::vector<SmallVector<int64_t, 4>> *lbs = nullptr,
H A DAffineStructures.h103 getHyperrectangular(ValueRange ivs, ValueRange lbs, ValueRange ubs);
/llvm-project-15.0.7/flang/lib/Evaluate/
H A Dfold-designator.cpp53 Shape lbs{GetLBOUNDs(context_, x.base())}; in FoldDesignator() local
54 if (auto lowerBounds{AsConstantExtents(context_, lbs)}) { in FoldDesignator()
209 Shape lbs{GetRawLowerBounds(context, entity)}; in OffsetToArrayRef() local
210 auto lower{AsConstantExtents(context, lbs)}; in OffsetToArrayRef()
/llvm-project-15.0.7/llvm/include/llvm/Analysis/
H A DLoopIterator.h193 po_iterator_storage(LoopBlocksTraversal &lbs) : LBT(lbs) {} in po_iterator_storage() argument
/llvm-project-15.0.7/mlir/lib/Conversion/SCFToGPU/
H A DSCFToGPU.cpp162 SmallVector<Value, 6> lbs; member
186 lbs.reserve(numLoops); in collectBounds()
204 lbs.push_back(lowerBound); in collectBounds()
260 auto *lbArgumentIt = lbs.begin(); in createLaunch()
/llvm-project-15.0.7/mlir/include/mlir/Dialect/Affine/IR/
H A DAffineOps.h453 ArrayRef<int64_t> lbs, ArrayRef<int64_t> ubs,
457 void buildAffineLoopNest(OpBuilder &builder, Location loc, ValueRange lbs,
/llvm-project-15.0.7/mlir/lib/Dialect/Affine/Transforms/
H A DLoopFusion.cpp911 std::vector<SmallVector<int64_t, 4>> lbs; in createPrivateMemRef() local
913 lbs.reserve(rank); in createPrivateMemRef()
917 region.getConstantBoundingSizeAndShape(&newShape, &lbs, &lbDivisors); in createPrivateMemRef()
931 assert(lbs[d].size() == cst->getNumCols() - rank && "incorrect bound size"); in createPrivateMemRef()
935 offset = offset + lbs[d][j] * top.getAffineDimExpr(j); in createPrivateMemRef()
939 (offset + lbs[d][cst->getNumCols() - 1 - rank]).floorDiv(lbDivisors[d]); in createPrivateMemRef()
/llvm-project-15.0.7/mlir/lib/Dialect/SCF/IR/
H A DSCF.cpp506 OpBuilder &builder, Location loc, ValueRange lbs, ValueRange ubs, in buildLoopNest() argument
510 assert(lbs.size() == ubs.size() && in buildLoopNest()
512 assert(lbs.size() == steps.size() && in buildLoopNest()
516 if (lbs.empty()) { in buildLoopNest()
531 loops.reserve(lbs.size()); in buildLoopNest()
532 ivs.reserve(lbs.size()); in buildLoopNest()
535 for (unsigned i = 0, e = lbs.size(); i < e; ++i) { in buildLoopNest()
537 currentLoc, lbs[i], ubs[i], steps[i], currentIterArgs, in buildLoopNest()
579 OpBuilder &builder, Location loc, ValueRange lbs, ValueRange ubs, in buildLoopNest() argument
583 return buildLoopNest(builder, loc, lbs, ubs, steps, llvm::None, in buildLoopNest()
/llvm-project-15.0.7/mlir/lib/Dialect/Affine/IR/
H A DAffineOps.cpp2157 OpBuilder &builder, Location loc, BoundListTy lbs, BoundListTy ubs, in buildAffineLoopNestImpl() argument
2161 assert(lbs.size() == ubs.size() && "Mismatch in number of arguments"); in buildAffineLoopNestImpl()
2162 assert(lbs.size() == steps.size() && "Mismatch in number of arguments"); in buildAffineLoopNestImpl()
2166 if (lbs.empty()) { in buildAffineLoopNestImpl()
2174 ivs.reserve(lbs.size()); in buildAffineLoopNestImpl()
2175 for (unsigned i = 0, e = lbs.size(); i < e; ++i) { in buildAffineLoopNestImpl()
2220 OpBuilder &builder, Location loc, ArrayRef<int64_t> lbs, in buildAffineLoopNest() argument
2223 buildAffineLoopNestImpl(builder, loc, lbs, ubs, steps, bodyBuilderFn, in buildAffineLoopNest()
2228 OpBuilder &builder, Location loc, ValueRange lbs, ValueRange ubs, in buildAffineLoopNest() argument
2231 buildAffineLoopNestImpl(builder, loc, lbs, ubs, steps, bodyBuilderFn, in buildAffineLoopNest()
[all …]
/llvm-project-15.0.7/mlir/lib/Dialect/Linalg/Transforms/
H A DTiling.cpp568 SmallVector<Value> lbs, dims, allDims, steps; in tilePadOp() local
572 lbs.push_back(ranges[i].offset); in tilePadOp()
581 builder, loc, lbs, /*ubs=*/dims, steps, ValueRange(destOperand), in tilePadOp()
/llvm-project-15.0.7/mlir/include/mlir/Dialect/Linalg/Utils/
H A DUtils.h224 ValueRange tileSizes, AffineMap map, ValueRange lbs,
/llvm-project-15.0.7/mlir/lib/Dialect/Affine/Utils/
H A DLoopUtils.cpp2087 std::vector<SmallVector<int64_t, 4>> lbs; in generateCopy() local
2089 lbs.reserve(rank); in generateCopy()
2091 &fastBufferShape, &lbs, &lbDivisors); in generateCopy()
2123 assert(lbs[d].size() == cst->getNumCols() - rank && "incorrect bound size"); in generateCopy()
2127 offset = offset + lbs[d][j] * top.getAffineDimExpr(j); in generateCopy()
2130 (offset + lbs[d][cst->getNumCols() - 1 - rank]).floorDiv(lbDivisors[d]); in generateCopy()
H A DLoopFusionUtils.cpp448 if (AffineMap lbMap = srcSlice.lbs[i]) { in fuseLoops()
/llvm-project-15.0.7/flang/lib/Optimizer/Builder/
H A DMutableBox.cpp846 auto lbs = in finalizeRealloc()
862 heap, lbs, extents, lengths); in finalizeRealloc()
/llvm-project-15.0.7/flang/lib/Lower/
H A DConvertExpr.cpp3756 auto lbs = fir::factory::getOrigins(arrayOperands[0].shape); in lowerAllocatableArrayAssignment() local
3757 lbounds.append(lbs.begin(), lbs.end()); in lowerAllocatableArrayAssignment()
4231 auto lbs = lbounds.value(); in convertElementForUpdate() local
4232 if (lbs.size() > 0) { in convertElementForUpdate()
4234 auto shiftTy = fir::ShiftType::get(eleTy.getContext(), lbs.size()); in convertElementForUpdate()
4235 mlir::Value shiftOp = builder.create<fir::ShiftOp>(loc, shiftTy, lbs); in convertElementForUpdate()
4240 auto lbs = lbounds.value(); in convertElementForUpdate() local
4241 if (lbs.size() > 0) { in convertElementForUpdate()
4243 auto shapeShiftArgs = flatZip(lbs, ubounds.value()); in convertElementForUpdate()
7395 const llvm::SmallVector<mlir::Value> &lbs, in setPointerAssignmentBounds() argument
[all …]
/llvm-project-15.0.7/flang/include/flang/Parser/
H A Dparse-tree.h3038 Suffix(LanguageBindingSpec &&lbs, std::optional<Name> &&rn) in Suffix()
3039 : binding(std::move(lbs)), resultName(std::move(rn)) {} in Suffix()
3040 Suffix(Name &&rn, std::optional<LanguageBindingSpec> &&lbs) in Suffix()
3041 : binding(std::move(lbs)), resultName(std::move(rn)) {} in Suffix()