diff options
Diffstat (limited to 'mlir/lib/Transforms/LoopFusion.cpp')
| -rw-r--r-- | mlir/lib/Transforms/LoopFusion.cpp | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/mlir/lib/Transforms/LoopFusion.cpp b/mlir/lib/Transforms/LoopFusion.cpp index b7b69fa54fe..0f39e52eefb 100644 --- a/mlir/lib/Transforms/LoopFusion.cpp +++ b/mlir/lib/Transforms/LoopFusion.cpp @@ -1006,9 +1006,9 @@ static Value *createPrivateMemRef(AffineForOp forOp, Operation *srcStoreOpInst, auto *forInst = forOp.getOperation(); // Create builder to insert alloc op just before 'forOp'. - FuncBuilder b(forInst); + OpBuilder b(forInst); // Builder to create constants at the top level. - FuncBuilder top(forInst->getFunction()); + OpBuilder top(forInst->getFunction()->getBody()); // Create new memref type based on slice bounds. auto *oldMemRef = cast<StoreOp>(srcStoreOpInst).getMemRef(); auto oldMemRefType = oldMemRef->getType().cast<MemRefType>(); |
