diff options
| author | Matthias Springer <me@m-sp.org> | 2025-07-27 08:35:51 +0000 |
|---|---|---|
| committer | Matthias Springer <me@m-sp.org> | 2025-07-27 09:59:13 +0000 |
| commit | b7bd595d950ba52d85882c7c844cb7f446169b06 (patch) | |
| tree | 35701217afc2e18f87dbadb0b49e4233dcad2258 /mlir/lib/Dialect/GPU/Transforms/AllReduceLowering.cpp | |
| parent | cf1abe67b9c6881527e42c1f6571665bb506b708 (diff) | |
[mlir][IR] Move insertion point when splitting blocks / moving opsusers/matthias-springer/ip_split_move
Diffstat (limited to 'mlir/lib/Dialect/GPU/Transforms/AllReduceLowering.cpp')
| -rw-r--r-- | mlir/lib/Dialect/GPU/Transforms/AllReduceLowering.cpp | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/mlir/lib/Dialect/GPU/Transforms/AllReduceLowering.cpp b/mlir/lib/Dialect/GPU/Transforms/AllReduceLowering.cpp index 8c449144af3a..0155b478f7cf 100644 --- a/mlir/lib/Dialect/GPU/Transforms/AllReduceLowering.cpp +++ b/mlir/lib/Dialect/GPU/Transforms/AllReduceLowering.cpp @@ -184,6 +184,7 @@ private: return [&body, this](Value lhs, Value rhs) -> Value { Block *block = rewriter.getInsertionBlock(); Block *split = rewriter.splitBlock(block, rewriter.getInsertionPoint()); + rewriter.setInsertionPointToEnd(block); // Insert accumulator body between split block. IRMapping mapping; |