Searched refs:depthToLoops (Results 1 – 4 of 4) sorted by relevance
| /external/llvm-project/mlir/test/lib/Transforms/ |
| D | TestLoopFusion.cpp | 149 static bool iterateLoops(ArrayRef<SmallVector<AffineForOp, 2>> depthToLoops, in iterateLoops() argument
152 for (unsigned loopDepth = 0, end = depthToLoops.size(); loopDepth < end; in iterateLoops()
154 auto &loops = depthToLoops[loopDepth]; in iterateLoops()
160 fn(loops[j], loops[k], j, k, loopDepth, depthToLoops.size()); in iterateLoops()
170 std::vector<SmallVector<AffineForOp, 2>> depthToLoops; in runOnFunction() local
174 depthToLoops.clear(); in runOnFunction()
176 gatherLoops(getFunction(), depthToLoops); in runOnFunction()
179 } while (iterateLoops(depthToLoops, testLoopFusionTransformation, in runOnFunction()
185 gatherLoops(getFunction(), depthToLoops); in runOnFunction()
189 iterateLoops(depthToLoops, testDependenceCheck); in runOnFunction()
[all …]
|
| /external/llvm-project/mlir/test/lib/Dialect/Affine/ |
| D | TestAffineDataCopy.cpp | 52 std::vector<SmallVector<AffineForOp, 2>> depthToLoops; in runOnFunction() local
53 gatherLoops(getFunction(), depthToLoops); in runOnFunction()
54 assert(depthToLoops.size() && "Loop nest not found"); in runOnFunction()
58 unsigned innermostLoopIdx = depthToLoops.size() - 1; in runOnFunction()
59 if (depthToLoops[0].size() != 1 || depthToLoops[innermostLoopIdx].size() != 1) in runOnFunction()
62 auto loopNest = depthToLoops[0][0]; in runOnFunction()
63 auto innermostLoop = depthToLoops[innermostLoopIdx][0]; in runOnFunction()
|
| /external/llvm-project/mlir/include/mlir/Transforms/ |
| D | LoopUtils.h | 286 std::vector<SmallVector<AffineForOp, 2>> &depthToLoops);
|
| /external/llvm-project/mlir/lib/Transforms/Utils/ |
| D | LoopUtils.cpp | 2876 std::vector<SmallVector<AffineForOp, 2>> &depthToLoops) { in gatherLoopsInBlock() argument
2878 assert(currLoopDepth <= depthToLoops.size() && "Unexpected currLoopDepth"); in gatherLoopsInBlock()
2879 if (currLoopDepth == depthToLoops.size()) in gatherLoopsInBlock()
2880 depthToLoops.push_back(SmallVector<AffineForOp, 2>()); in gatherLoopsInBlock()
2884 depthToLoops[currLoopDepth].push_back(forOp); in gatherLoopsInBlock()
2885 gatherLoopsInBlock(forOp.getBody(), currLoopDepth + 1, depthToLoops); in gatherLoopsInBlock()
2892 std::vector<SmallVector<AffineForOp, 2>> &depthToLoops) { in gatherLoops() argument
2894 gatherLoopsInBlock(&block, /*currLoopDepth=*/0, depthToLoops); in gatherLoops()
2897 if (!depthToLoops.empty()) { in gatherLoops()
2898 assert(depthToLoops.back().empty() && "Last loop level is not empty?"); in gatherLoops()
[all …]
|