Lines Matching refs:srcForOp
137 static Operation *getFusedLoopNestInsertionPoint(AffineForOp srcForOp, in getFusedLoopNestInsertionPoint() argument
140 srcForOp->isBeforeInBlock(dstForOp.getOperation()); in getFusedLoopNestInsertionPoint()
141 auto forOpA = isSrcForOpBeforeDstForOp ? srcForOp : dstForOp; in getFusedLoopNestInsertionPoint()
142 auto forOpB = isSrcForOpBeforeDstForOp ? dstForOp : srcForOp; in getFusedLoopNestInsertionPoint()
263 FusionResult mlir::canFuseLoops(AffineForOp srcForOp, AffineForOp dstForOp, in canFuseLoops() argument
273 auto *block = srcForOp->getBlock(); in canFuseLoops()
281 if (!getFusedLoopNestInsertionPoint(srcForOp, dstForOp)) { in canFuseLoops()
288 srcForOp->isBeforeInBlock(dstForOp.getOperation()); in canFuseLoops()
290 auto forOpA = isSrcForOpBeforeDstForOp ? srcForOp : dstForOp; in canFuseLoops()
291 auto forOpB = isSrcForOpBeforeDstForOp ? dstForOp : srcForOp; in canFuseLoops()
322 *srcForOp.getOperation(), *dstForOp.getOperation()); in canFuseLoops()
367 void mlir::fuseLoops(AffineForOp srcForOp, AffineForOp dstForOp, in fuseLoops() argument
372 b.clone(*srcForOp, mapper); in fuseLoops()
565 bool mlir::getFusionComputeCost(AffineForOp srcForOp, LoopNestStats &srcStats, in getFusionComputeCost() argument
588 srcForOp.walk([&](Operation *op) { in getFusionComputeCost()
621 srcForOp.getOperation(), srcStats, &sliceTripCountMap, &computeCostMap); in getFusionComputeCost()