diff options
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; |