[Mlir-commits] [mlir] Refactor LoopFuseSiblingOp and support parallel fusion (PR #94391)

llvmlistbot at llvm.org llvmlistbot at llvm.org
Wed Jun 26 11:09:20 PDT 2024


================
@@ -57,4 +57,24 @@ class HasParallelRegion : public TraitBase<ConcreteType, HasParallelRegion> {
 /// Include the generated interface declarations.
 #include "mlir/Interfaces/LoopLikeInterface.h.inc"
 
+namespace mlir {
+/// A function that rewrites `target`'s terminator as a teminator obtained by
+/// fusing `source` into `target`.
+using FuseTerminatorFn =
+    std::function<void(RewriterBase &rewriter, LoopLikeOpInterface source,
----------------
srcarroll wrote:

ok. although that is the case now, i'm not sure if that will always be the case.  I figured its usage would be similar to `NewYieldValuesFn`, but will defer to reviewers on this.

https://github.com/llvm/llvm-project/pull/94391


More information about the Mlir-commits mailing list