Searched refs:dstForOp (Results 1 – 4 of 4) sorted by relevance
| /llvm-project-15.0.7/mlir/lib/Dialect/Affine/Utils/ |
| H A D | LoopFusionUtils.cpp | 141 srcForOp->isBeforeInBlock(dstForOp.getOperation()); in getFusedLoopNestInsertionPoint()
142 auto forOpA = isSrcForOpBeforeDstForOp ? srcForOp : dstForOp; in getFusedLoopNestInsertionPoint()
143 auto forOpB = isSrcForOpBeforeDstForOp ? dstForOp : srcForOp; in getFusedLoopNestInsertionPoint()
271 if (block != dstForOp->getBlock()) { in canFuseLoops()
278 if (!getFusedLoopNestInsertionPoint(srcForOp, dstForOp)) { in canFuseLoops()
285 srcForOp->isBeforeInBlock(dstForOp.getOperation()); in canFuseLoops()
287 auto forOpA = isSrcForOpBeforeDstForOp ? srcForOp : dstForOp; in canFuseLoops()
288 auto forOpB = isSrcForOpBeforeDstForOp ? dstForOp : srcForOp; in canFuseLoops()
319 *srcForOp.getOperation(), *dstForOp.getOperation()); in canFuseLoops()
432 void mlir::fuseLoops(AffineForOp srcForOp, AffineForOp dstForOp, in fuseLoops() argument
[all …]
|
| /llvm-project-15.0.7/mlir/include/mlir/Dialect/Affine/ |
| H A D | LoopFusionUtils.h | 113 canFuseLoops(AffineForOp srcForOp, AffineForOp dstForOp, unsigned dstLoopDepth,
121 void fuseLoops(AffineForOp srcForOp, AffineForOp dstForOp,
158 AffineForOp dstForOp, LoopNestStats &dstStats,
|
| /llvm-project-15.0.7/mlir/test/lib/Dialect/Affine/ |
| H A D | TestLoopFusion.cpp | 60 static bool testDependenceCheck(AffineForOp srcForOp, AffineForOp dstForOp, in testDependenceCheck() argument
66 mlir::canFuseLoops(srcForOp, dstForOp, d, &sliceUnion); in testDependenceCheck()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Affine/Transforms/ |
| H A D | LoopFusion.cpp | 1089 AffineForOp dstForOp, in isFusionProfitable() argument
1097 llvm::dbgs() << dstForOp << "\n"; in isFusionProfitable()
1116 if (!getLoopNestStats(dstForOp, &dstLoopNestStats)) in isFusionProfitable()
1150 uint64_t dstLoopNestCost = getComputeCost(dstForOp, dstLoopNestStats); in isFusionProfitable()
1161 if (!getFusionComputeCost(srcLoopIVs[0], srcLoopNestStats, dstForOp, in isFusionProfitable()
1260 auto dstMemSize = getMemoryFootprintBytes(dstForOp); in isFusionProfitable()
|