Searched refs:depthToLoops (Results 1 – 4 of 4) sorted by relevance
| /llvm-project-15.0.7/mlir/test/lib/Dialect/Affine/ |
| H A D | TestLoopFusion.cpp | 160 static bool iterateLoops(ArrayRef<SmallVector<AffineForOp, 2>> depthToLoops, in iterateLoops() argument
163 for (unsigned loopDepth = 0, end = depthToLoops.size(); loopDepth < end; in iterateLoops()
165 auto &loops = depthToLoops[loopDepth]; in iterateLoops()
171 fn(loops[j], loops[k], j, k, loopDepth, depthToLoops.size()); in iterateLoops()
181 std::vector<SmallVector<AffineForOp, 2>> depthToLoops; in runOnOperation() local
185 depthToLoops.clear(); in runOnOperation()
187 gatherLoops(getOperation(), depthToLoops); in runOnOperation()
190 } while (iterateLoops(depthToLoops, testLoopFusionTransformation, in runOnOperation()
196 gatherLoops(getOperation(), depthToLoops); in runOnOperation()
200 iterateLoops(depthToLoops, testDependenceCheck); in runOnOperation()
[all …]
|
| H A D | TestAffineDataCopy.cpp | 61 std::vector<SmallVector<AffineForOp, 2>> depthToLoops; in runOnOperation() local
62 gatherLoops(getOperation(), depthToLoops); in runOnOperation()
63 assert(!depthToLoops.empty() && "Loop nest not found"); in runOnOperation()
67 unsigned innermostLoopIdx = depthToLoops.size() - 1; in runOnOperation()
68 if (depthToLoops[0].size() != 1 || depthToLoops[innermostLoopIdx].size() != 1) in runOnOperation()
71 auto loopNest = depthToLoops[0][0]; in runOnOperation()
72 auto innermostLoop = depthToLoops[innermostLoopIdx][0]; in runOnOperation()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Affine/Utils/ |
| H A D | LoopUtils.cpp | 2596 std::vector<SmallVector<AffineForOp, 2>> &depthToLoops) { in gatherLoopsInBlock() argument
2598 assert(currLoopDepth <= depthToLoops.size() && "Unexpected currLoopDepth"); in gatherLoopsInBlock()
2599 if (currLoopDepth == depthToLoops.size()) in gatherLoopsInBlock()
2600 depthToLoops.emplace_back(); in gatherLoopsInBlock()
2604 depthToLoops[currLoopDepth].push_back(forOp); in gatherLoopsInBlock()
2605 gatherLoopsInBlock(forOp.getBody(), currLoopDepth + 1, depthToLoops); in gatherLoopsInBlock()
2612 std::vector<SmallVector<AffineForOp, 2>> &depthToLoops) { in gatherLoops() argument
2614 gatherLoopsInBlock(&block, /*currLoopDepth=*/0, depthToLoops); in gatherLoops()
2617 if (!depthToLoops.empty()) { in gatherLoops()
2618 assert(depthToLoops.back().empty() && "Last loop level is not empty?"); in gatherLoops()
[all …]
|
| /llvm-project-15.0.7/mlir/include/mlir/Dialect/Affine/ |
| H A D | LoopUtils.h | 267 std::vector<SmallVector<AffineForOp, 2>> &depthToLoops);
|