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

llvmlistbot at llvm.org llvmlistbot at llvm.org
Wed Jun 26 18:08:24 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:

i got crashes when changing to `function_ref` for `NewYieldValuesFn`, but didn't really look into it so just keeping as is

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


More information about the Mlir-commits mailing list