[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