Searched refs:consumerOp (Results 1 – 3 of 3) sorted by relevance
| /llvm-project-15.0.7/mlir/lib/Dialect/Linalg/Transforms/ |
| H A D | Fusion.cpp | 283 LinalgOp consumerOp = dyn_cast<LinalgOp>(consumerOpOperand.getOwner()); in findFusableProducer() local
284 if (!consumerOp) in findFusableProducer()
293 << "Dependencies into: " << *consumerOp.getOperation() << "\n"); in findFusableProducer()
295 dependenceGraph.getDependencesInto(consumerOp, depType), in findFusableProducer()
319 if (producer.hasBufferSemantics() && consumerOp.hasBufferSemantics() && in findFusableProducer()
320 isFusableInto(dependenceGraph, consumerOp, consumerOpOperand.get(), in findFusableProducer()
323 if (producer.hasTensorSemantics() && consumerOp.hasTensorSemantics()) { in findFusableProducer()
422 LinalgOp consumerOp = dyn_cast<LinalgOp>(consumerOpOperand.getOwner()); in fuseProducerOfTensor() local
423 if (!consumerOp) in fuseProducerOfTensor()
443 b.setInsertionPoint(consumerOp); in fuseProducerOfTensor()
[all …]
|
| H A D | FusionOnTensors.cpp | 41 LinalgOp consumerOp = consumerOperand->getOwner(); in getTiledSliceDims() local
42 AffineMap indexingMap = consumerOp.getTiedIndexingMap(consumerOperand); in getTiledSliceDims()
348 LinalgOp consumerOp = consumerOpOperand->getOwner(); in fuseProducer() local
350 consumerOp->getBlock() != rootOp->getBlock()) in fuseProducer()
355 BlockArgument bbArg = consumerOp.getTiedBlockArgument(consumerOpOperand); in fuseProducer()
374 consumerOpOperand, tiledRootAndFusedOpsLoops[consumerOp]); in fuseProducer()
424 OpBuilder &b, LinalgOp consumerOp, ArrayRef<int64_t> tileSizes, in tileConsumerAndFuseProducers() argument
433 TileLoopNest tileLoopNest(consumerOp); in tileConsumerAndFuseProducers()
438 llvm::to_vector<6>(consumerOp.iterator_types().getAsRange<StringAttr>()); in tileConsumerAndFuseProducers()
|
| /llvm-project-15.0.7/mlir/include/mlir/Dialect/Linalg/Utils/ |
| H A D | Utils.h | 443 OpBuilder &b, LinalgOp consumerOp, ArrayRef<int64_t> tileSizes,
|