Lines Matching refs:srcForOp
134 static Operation *getFusedLoopNestInsertionPoint(AffineForOp srcForOp, in getFusedLoopNestInsertionPoint() argument
136 bool isSrcForOpBeforeDstForOp = srcForOp->isBeforeInBlock(dstForOp); in getFusedLoopNestInsertionPoint()
137 auto forOpA = isSrcForOpBeforeDstForOp ? srcForOp : dstForOp; in getFusedLoopNestInsertionPoint()
138 auto forOpB = isSrcForOpBeforeDstForOp ? dstForOp : srcForOp; in getFusedLoopNestInsertionPoint()
249 FusionResult mlir::affine::canFuseLoops(AffineForOp srcForOp, in canFuseLoops() argument
260 auto *block = srcForOp->getBlock(); in canFuseLoops()
268 if (!getFusedLoopNestInsertionPoint(srcForOp, dstForOp)) { in canFuseLoops()
274 bool isSrcForOpBeforeDstForOp = srcForOp->isBeforeInBlock(dstForOp); in canFuseLoops()
276 auto forOpA = isSrcForOpBeforeDstForOp ? srcForOp : dstForOp; in canFuseLoops()
277 auto forOpB = isSrcForOpBeforeDstForOp ? dstForOp : srcForOp; in canFuseLoops()
308 affine::getNumCommonSurroundingLoops(*srcForOp, *dstForOp); in canFuseLoops()
426 void mlir::affine::fuseLoops(AffineForOp srcForOp, AffineForOp dstForOp, in fuseLoops() argument
432 b.clone(*srcForOp, mapper); in fuseLoops()
575 bool mlir::affine::getFusionComputeCost(AffineForOp srcForOp, in getFusionComputeCost() argument
599 srcForOp.walk([&](AffineWriteOpInterface storeOp) { in getFusionComputeCost()
629 srcForOp, srcStats, &sliceTripCountMap, &computeCostMap); in getFusionComputeCost()