Home
last modified time | relevance | path

Searched refs:srcForOp (Results 1 – 4 of 4) sorted by relevance

/llvm-project-15.0.7/mlir/lib/Dialect/Affine/Utils/
H A DLoopFusionUtils.cpp141 srcForOp->isBeforeInBlock(dstForOp.getOperation()); in getFusedLoopNestInsertionPoint()
142 auto forOpA = isSrcForOpBeforeDstForOp ? srcForOp : dstForOp; in getFusedLoopNestInsertionPoint()
143 auto forOpB = isSrcForOpBeforeDstForOp ? dstForOp : srcForOp; in getFusedLoopNestInsertionPoint()
270 auto *block = srcForOp->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()
438 b.clone(*srcForOp, mapper); in fuseLoops()
[all …]
H A DLoopUtils.cpp184 unsigned offset, AffineForOp srcForOp, OpBuilder b) { in generateShiftedLoop() argument
185 auto lbOperands = srcForOp.getLowerBoundOperands(); in generateShiftedLoop()
186 auto ubOperands = srcForOp.getUpperBoundOperands(); in generateShiftedLoop()
191 auto loopChunk = b.create<AffineForOp>(srcForOp.getLoc(), lbOperands, lbMap, in generateShiftedLoop()
192 ubOperands, ubMap, srcForOp.getStep()); in generateShiftedLoop()
194 auto srcIV = srcForOp.getInductionVar(); in generateShiftedLoop()
208 srcForOp.getLoc(), in generateShiftedLoop()
210 -static_cast<int64_t>(srcForOp.getStep() * shift)), in generateShiftedLoop()
/llvm-project-15.0.7/mlir/include/mlir/Dialect/Affine/
H A DLoopFusionUtils.h113 canFuseLoops(AffineForOp srcForOp, AffineForOp dstForOp, unsigned dstLoopDepth,
121 void fuseLoops(AffineForOp srcForOp, AffineForOp dstForOp,
157 bool getFusionComputeCost(AffineForOp srcForOp, LoopNestStats &srcStats,
/llvm-project-15.0.7/mlir/test/lib/Dialect/Affine/
H A DTestLoopFusion.cpp60 static bool testDependenceCheck(AffineForOp srcForOp, AffineForOp dstForOp, in testDependenceCheck() argument
66 mlir::canFuseLoops(srcForOp, dstForOp, d, &sliceUnion); in testDependenceCheck()
68 srcForOp->emitRemark("block-level dependence preventing" in testDependenceCheck()