[Mlir-commits] [mlir] [MLIR][SCF] Add an API to fuse consumer to a producer within scf loop (PR #88712)
llvmlistbot at llvm.org
llvmlistbot at llvm.org
Thu May 16 23:55:59 PDT 2024
================
@@ -239,6 +240,18 @@ tileConsumerAndFuseProducersUsingSCF(RewriterBase &rewriter,
TilingInterface consumer,
const SCFTileAndFuseOptions &options);
+/// Fuse the consumer of the source of `candidateSliceOp` by computing the
+/// required slice of the consumer in-place. Note that the method
+/// replaces the uses of `candidateSliceOp` with the tiled and fused consumer
+/// value but does not delete the slice operation.
+struct SCFFuseConsumerOfSliceResult {
+ Operation *origConsumer; // Original untiled consumer.
+ Operation *tiledAndFusedConsumer; // Tiled and fused consumer op.
----------------
MaheshRavishankar wrote:
Not sure this make sense, but similarly to be consistent, this should be an `OpOperand *` pointing the operand in the newly created (i.e tiled and fused) consumer.
https://github.com/llvm/llvm-project/pull/88712
More information about the Mlir-commits
mailing list