Skip to content

Commit 74437c4

Browse files
committed
extend consumer fuse to nested scf loop (v2)
1 parent a841446 commit 74437c4

File tree

3 files changed

+450
-40
lines changed

3 files changed

+450
-40
lines changed

mlir/include/mlir/Dialect/SCF/Transforms/TileUsingInterface.h

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -254,6 +254,10 @@ struct SCFFuseConsumerOfSliceResult {
254254
*tiledAndFusedConsumerOperand; // Tiled and fused consumer's operand.
255255
SmallVector<Operation *> tiledOps;
256256
};
257+
FailureOr<scf::SCFFuseConsumerOfSliceResult>
258+
tileAndFuseConsumerOfSliceImpl(RewriterBase &rewriter,
259+
Operation *candidateSliceOp);
260+
257261
FailureOr<scf::SCFFuseConsumerOfSliceResult>
258262
tileAndFuseConsumerOfSlice(RewriterBase &rewriter, Operation *candidateSliceOp);
259263

0 commit comments

Comments
 (0)