diff options
Diffstat (limited to 'mlir/lib/Transforms/PipelineDataTransfer.cpp')
| -rw-r--r-- | mlir/lib/Transforms/PipelineDataTransfer.cpp | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/mlir/lib/Transforms/PipelineDataTransfer.cpp b/mlir/lib/Transforms/PipelineDataTransfer.cpp index e72b9ef80df..0019714b6a3 100644 --- a/mlir/lib/Transforms/PipelineDataTransfer.cpp +++ b/mlir/lib/Transforms/PipelineDataTransfer.cpp @@ -121,8 +121,8 @@ static bool doubleBuffer(Value *oldMemRef, ForInst *forInst) { int64_t step = forInst->getStep(); auto modTwoMap = bInner.getAffineMap(/*dimCount=*/1, /*symbolCount=*/0, {d0.floorDiv(step) % 2}, {}); - auto ivModTwoOp = - bInner.create<AffineApplyOp>(forInst->getLoc(), modTwoMap, forInst); + auto ivModTwoOp = bInner.create<AffineApplyOp>(forInst->getLoc(), modTwoMap, + forInst->getInductionVar()); // replaceAllMemRefUsesWith will always succeed unless the forInst body has // non-deferencing uses of the memref. |
