| /llvm-project-15.0.7/mlir/lib/Dialect/Linalg/Transforms/ |
| H A D | Tiling.cpp | 130 SmallVector<Value, 4> loopRanges = in computeMultiTileSizes() local
132 Value tripCount = loopRanges[dimension]; in computeMultiTileSizes()
226 if (loopRanges.empty()) in tileToForeachThreadOpImpl()
229 if (llvm::any_of(loopRanges, hasStrideOne)) in tileToForeachThreadOpImpl()
259 int64_t nLoops = loopRanges.size(); in tileToForeachThreadOpImpl()
269 tiledSizes.push_back(loopRanges[loopIdx].size); in tileToForeachThreadOpImpl()
277 Value size = loopRanges[loopIdx].size; in tileToForeachThreadOpImpl()
278 Value offset = loopRanges[loopIdx].offset; in tileToForeachThreadOpImpl()
351 unsigned nLoops = loopRanges.size(); in tileToForeachThreadOpUsingTileSizes()
404 SmallVector<Range, 4> loopRanges; in tileLinalgOpImpl() local
[all …]
|
| H A D | Fusion.cpp | 117 SmallVector<Range, 8> loopRanges; in fuse() local
130 loopRanges.push_back(it->second); in fuse()
132 << loopRanges.back() << "\n"); in fuse()
135 loopRanges.push_back(Range{zero, dim, one}); in fuse()
137 << loopRanges.back() << "\n"); in fuse()
171 llvm::transform(loopRanges, std::back_inserter(allIvs), in fuse()
|
| H A D | SplitReduction.cpp | 97 SmallVector<int64_t, 4> loopRanges = op.getStaticLoopRanges(); in splitReduction() local
98 int64_t reductionDimSize = loopRanges[reductionDim]; in splitReduction()
101 insertSplitDimension >= loopRanges.size()) in splitReduction()
290 SmallVector<int64_t> loopRanges = op.getStaticLoopRanges(); in splitReductionByScaling() local
291 int64_t reductionDimSize = loopRanges[reductionDimPos]; in splitReductionByScaling()
294 insertSplitDimension >= loopRanges.size()) in splitReductionByScaling()
|
| H A D | Loops.cpp | 211 auto loopRanges = linalgOp.createLoopRanges(rewriter, linalgOp.getLoc()); in linalgOpToLoopsImpl() local
216 rewriter, linalgOp.getLoc(), loopRanges, linalgOp, iteratorTypes, in linalgOpToLoopsImpl()
|
| H A D | ElementwiseOpFusion.cpp | 1393 SmallVector<Range> loopRanges = in collapseGenericOpIterationDims() local
1396 assert(llvm::all_of(loopRanges, in collapseGenericOpIterationDims()
1403 llvm::map_range(loopRanges, [](Range range) { return range.size; })); in collapseGenericOpIterationDims()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/SCF/Transforms/ |
| H A D | TileUsingInterface.cpp | 100 ArrayRef<Range> loopRanges, ArrayRef<Value> tileSizeVals, in generateTileLoopNest() argument
103 assert(!loopRanges.empty() && "expected at least one loop range"); in generateTileLoopNest()
104 assert(loopRanges.size() == tileSizeVals.size() && in generateTileLoopNest()
108 offsets.resize(loopRanges.size()); in generateTileLoopNest()
109 sizes.resize(loopRanges.size()); in generateTileLoopNest()
119 for (auto loopRange : llvm::enumerate(loopRanges)) { in generateTileLoopNest()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Tensor/IR/ |
| H A D | TensorTilingInterfaceImpl.cpp | 55 SmallVector<Range> loopRanges(reifiedShapes[0].size(), {zero, one, one}); in getIterationDomain() local
57 loopRanges[ub.index()].size = ub.value(); in getIterationDomain()
58 return loopRanges; in getIterationDomain()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Linalg/Utils/ |
| H A D | Utils.cpp | 501 OpBuilder &b, Location loc, ArrayRef<Range> loopRanges, LinalgOp linalgOp, in doit() argument
517 parallelLoopRanges.push_back(loopRanges[iteratorType.index()]); in doit()
527 unpackRanges(loopRanges, lbs, ubs, steps); in doit()
561 OpBuilder &b, Location loc, ArrayRef<Range> loopRanges, LinalgOp linalgOp, in doit() argument
570 unpackRanges(loopRanges, lbs, ubs, steps); in doit()
726 OpBuilder &b, Location loc, ArrayRef<Range> loopRanges, LinalgOp linalgOp, in doit() argument
736 assert(iteratorTypes.size() >= loopRanges.size() && in doit()
738 iteratorTypes = iteratorTypes.take_front(loopRanges.size()); in doit()
747 unpackRanges(loopRanges, lbsStorage, ubsStorage, stepsStorage); in doit()
758 parallelLoopRanges.push_back(loopRanges[iteratorType.index()]); in doit()
|
| /llvm-project-15.0.7/mlir/include/mlir/Dialect/Linalg/Utils/ |
| H A D | Utils.h | 481 static void doit(OpBuilder &b, Location loc, ArrayRef<Range> loopRanges,
|