Lines Matching refs:srcForOp
138 static Operation *getFusedLoopNestInsertionPoint(AffineForOp srcForOp, in getFusedLoopNestInsertionPoint() argument
141 srcForOp->isBeforeInBlock(dstForOp.getOperation()); in getFusedLoopNestInsertionPoint()
142 auto forOpA = isSrcForOpBeforeDstForOp ? srcForOp : dstForOp; in getFusedLoopNestInsertionPoint()
143 auto forOpB = isSrcForOpBeforeDstForOp ? dstForOp : srcForOp; in getFusedLoopNestInsertionPoint()
260 FusionResult mlir::canFuseLoops(AffineForOp srcForOp, AffineForOp dstForOp, in canFuseLoops() argument
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()
432 void mlir::fuseLoops(AffineForOp srcForOp, AffineForOp dstForOp, in fuseLoops() argument
438 b.clone(*srcForOp, mapper); in fuseLoops()
580 bool mlir::getFusionComputeCost(AffineForOp srcForOp, LoopNestStats &srcStats, in getFusionComputeCost() argument
603 srcForOp.walk([&](Operation *op) { in getFusionComputeCost()
636 srcForOp.getOperation(), srcStats, &sliceTripCountMap, &computeCostMap); in getFusionComputeCost()