Home
last modified time | relevance | path

Searched refs:tileLoopNest (Results 1 – 3 of 3) sorted by relevance

/llvm-project-15.0.7/mlir/lib/Dialect/Linalg/Transforms/
H A DFusionOnTensors.cpp433 TileLoopNest tileLoopNest(consumerOp); in tileConsumerAndFuseProducers() local
450 tileLoopNest.fuseProducer(b, candidates.pop_back_val()); in tileConsumerAndFuseProducers()
469 if (failed(tileLoopNest.tileRootOp(b, outerTileSizes, tileInterchange, in tileConsumerAndFuseProducers()
472 fuseProducersGreedily(tileLoopNest.getRootOp().getOutputOperands()); in tileConsumerAndFuseProducers()
475 if (failed(tileLoopNest.tileRootOp(b, innerTileSizes, tileInterchange, in tileConsumerAndFuseProducers()
478 fuseProducersGreedily(tileLoopNest.getRootOp().getInputOperands()); in tileConsumerAndFuseProducers()
481 if (tileLoopNest.isEmpty()) in tileConsumerAndFuseProducers()
484 return tileLoopNest; in tileConsumerAndFuseProducers()
H A DTransforms.cpp518 FailureOr<TileLoopNest> tileLoopNest = in returningMatchAndRewrite() local
521 if (failed(tileLoopNest)) in returningMatchAndRewrite()
526 rootOp->replaceAllUsesWith(tileLoopNest->getRootOpReplacementResults()); in returningMatchAndRewrite()
529 for (LinalgOp linalgOp : tileLoopNest->getAllTiledAndFusedOps()) in returningMatchAndRewrite()
531 return tileLoopNest; in returningMatchAndRewrite()
/llvm-project-15.0.7/mlir/lib/Dialect/Linalg/TransformOps/
H A DLinalgTransformOps.cpp178 FailureOr<TileLoopNest> tileLoopNest = in apply() local
180 if (failed(tileLoopNest)) in apply()
184 tiledLinalgOp.op = tileLoopNest->getRootOp(); in apply()
185 tiledLinalgOp.loops = {tileLoopNest->getLoopOps().begin(), in apply()
186 tileLoopNest->getLoopOps().end()}; in apply()