| /llvm-project-15.0.7/mlir/lib/Dialect/GPU/Transforms/ |
| H A D | MemoryPromotion.cpp | 36 SmallVector<Value, 4> lbs, ubs, steps; in insertCopyLoops() local
45 ubs.resize(extraLoops, one); in insertCopyLoops()
51 ubs.reserve(lbs.size()); in insertCopyLoops()
54 ubs.push_back(b.createOrFold<memref::DimOp>( in insertCopyLoops()
70 b, b.getLoc(), lbs, ubs, steps, in insertCopyLoops()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Linalg/Utils/ |
| H A D | Utils.cpp | 133 SmallVectorImpl<Value> &ubs, in unpackRanges() argument
137 ubs.emplace_back(range.size); in unpackRanges()
526 SmallVector<Value, 4> lbs, ubs, steps; in doit() local
527 unpackRanges(loopRanges, lbs, ubs, steps); in doit()
529 b, loc, lbs, ubs, steps, iterArgInitValues, 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());
697 Value cond = ab.slt(lbs[0], ubs[0]);
699 cond = ab._and(cond, ab.slt(lbs[i], ubs[i]));
[all …]
|
| /llvm-project-15.0.7/mlir/include/mlir/Dialect/SCF/IR/ |
| H A D | SCF.h | 89 OpBuilder &builder, Location loc, ValueRange lbs, ValueRange ubs,
99 ValueRange ubs, ValueRange steps,
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Affine/Analysis/ |
| H A D | Utils.cpp | 111 LogicalResult ret = cst->addSliceBounds(ivs, lbs, ubs, lbOperands[0]); in getAsConstraints()
121 ubs.clear(); in clearBounds()
140 for (auto &en : llvm::enumerate(ubs)) { in dump()
154 assert(lbs.size() == ubs.size() && !lbs.empty() && !ivs.empty() && in isSliceMaximalFastCheck()
159 AffineMap ubMap = ubs[i]; in isSliceMaximalFastCheck()
932 sliceUnion->ubs.resize(numSliceLoopIVs, AffineMap()); in computeSliceUnion()
937 &sliceUnion->ubs); in computeSliceUnion()
1002 AffineMap ubMap = slice.ubs[i]; in buildSliceTripCountMap()
1076 sliceState->ubs.resize(numSliceLoopIVs, AffineMap()); in getComputationSliceState()
1141 sliceState->ubs[j] = AffineMap(); in getComputationSliceState()
[all …]
|
| H A D | AffineStructures.cpp | 199 ValueRange ubs) { in getHyperrectangular() argument
203 assert(nIvs == ubs.size() && "expected as many upper bounds as ivs"); in getHyperrectangular()
210 unsigned ubsStart = res.appendDimVar(ubs); in getHyperrectangular()
|
| /llvm-project-15.0.7/mlir/test/lib/Dialect/Affine/ |
| H A D | TestLoopFusion.cpp | 96 assert(sliceUnion.lbs.size() == sliceUnion.ubs.size()); in getSliceStr()
102 sliceUnion.ubs[k].print(os); in getSliceStr()
|
| /llvm-project-15.0.7/mlir/include/mlir/Dialect/Affine/IR/ |
| H A D | AffineOps.h | 453 ArrayRef<int64_t> lbs, ArrayRef<int64_t> ubs,
458 ValueRange ubs, ArrayRef<int64_t> steps,
|
| /llvm-project-15.0.7/mlir/include/mlir/Dialect/Affine/Analysis/ |
| H A D | Utils.h | 82 SmallVector<AffineMap, 4> ubs; member
|
| H A D | AffineStructures.h | 103 getHyperrectangular(ValueRange ivs, ValueRange lbs, ValueRange ubs);
|
| /llvm-project-15.0.7/mlir/include/mlir/Dialect/Linalg/Utils/ |
| H A D | Utils.h | 225 ValueRange ubs, ValueRange subShapeSizes,
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Affine/IR/ |
| H A D | AffineOps.cpp | 2157 OpBuilder &builder, Location loc, BoundListTy lbs, BoundListTy ubs, in buildAffineLoopNestImpl() argument
2161 assert(lbs.size() == ubs.size() && "Mismatch in number of arguments"); in buildAffineLoopNestImpl()
2190 auto loop = loopCreatorFn(builder, loc, lbs[i], ubs[i], steps[i], loopBody); in buildAffineLoopNestImpl()
2221 ArrayRef<int64_t> ubs, ArrayRef<int64_t> steps, 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()
3236 auto ubs = llvm::to_vector<4>(llvm::map_range(ranges, [&](int64_t value) { in build() local
3240 build(builder, result, resultTypes, reductions, lbs, /*lbArgs=*/{}, ubs, in build()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/SCF/IR/ |
| H A D | SCF.cpp | 506 OpBuilder &builder, Location loc, ValueRange lbs, ValueRange ubs, in buildLoopNest() argument
510 assert(lbs.size() == ubs.size() && 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/Utils/ |
| H A D | LoopFusionUtils.cpp | 453 if (AffineMap ubMap = srcSlice.ubs[i]) { in fuseLoops()
|
| /llvm-project-15.0.7/flang/lib/Lower/ |
| H A D | ConvertExpr.cpp | 7396 llvm::Optional<llvm::SmallVector<mlir::Value>> ubs) { in setPointerAssignmentBounds() argument
7398 ubounds = ubs; in setPointerAssignmentBounds()
|