File tree Expand file tree Collapse file tree 2 files changed +5
-4
lines changed
include/mlir/Dialect/SCF/Utils Expand file tree Collapse file tree 2 files changed +5
-4
lines changed Original file line number Diff line number Diff line change @@ -54,7 +54,7 @@ using NewYieldValueFn = std::function<SmallVector<Value>(
54
54
OpBuilder &b, Location loc, ArrayRef<BlockArgument> newBBArgs)>;
55
55
scf::ForOp replaceLoopWithNewYields (OpBuilder &builder, scf::ForOp loop,
56
56
ValueRange newIterOperands,
57
- NewYieldValueFn newYieldValuesFn);
57
+ const NewYieldValueFn & newYieldValuesFn);
58
58
59
59
// / Outline a region with a single block into a new FuncOp.
60
60
// / Assumes the FuncOp result types is the type of the yielded operands of the
Original file line number Diff line number Diff line change @@ -36,9 +36,10 @@ struct LoopParams {
36
36
};
37
37
} // namespace
38
38
39
- scf::ForOp mlir::replaceLoopWithNewYields (OpBuilder &builder, scf::ForOp loop,
40
- ValueRange newIterOperands,
41
- NewYieldValueFn newYieldValuesFn) {
39
+ scf::ForOp
40
+ mlir::replaceLoopWithNewYields (OpBuilder &builder, scf::ForOp loop,
41
+ ValueRange newIterOperands,
42
+ const NewYieldValueFn &newYieldValuesFn) {
42
43
// Create a new loop before the existing one, with the extra operands.
43
44
OpBuilder::InsertionGuard g (builder);
44
45
builder.setInsertionPoint (loop);
You can’t perform that action at this time.
0 commit comments