diff options
| author | Nicolas Vasilache <ntv@google.com> | 2018-10-09 16:39:24 -0700 |
|---|---|---|
| committer | jpienaar <jpienaar@google.com> | 2019-03-29 13:26:24 -0700 |
| commit | 1d3e7e2616b792831f0fc2db632eb1cad5241bfd (patch) | |
| tree | 59c93c29c00ede22bd8f70f541bc04019b4fd1c9 /mlir/lib/Transforms/PipelineDataTransfer.cpp | |
| parent | 82e55750d2dee6b927061574a31ed3eab2d92b16 (diff) | |
| download | bcm5719-llvm-1d3e7e2616b792831f0fc2db632eb1cad5241bfd.tar.gz bcm5719-llvm-1d3e7e2616b792831f0fc2db632eb1cad5241bfd.zip | |
[MLIR] AffineMap value type
This CL applies the same pattern as AffineExpr to AffineMap: a simple struct
that acts as the storage is allocated in the bump pointer. The AffineMap is
immutable and accessed everywhere by value.
PiperOrigin-RevId: 216445930
Diffstat (limited to 'mlir/lib/Transforms/PipelineDataTransfer.cpp')
| -rw-r--r-- | mlir/lib/Transforms/PipelineDataTransfer.cpp | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/mlir/lib/Transforms/PipelineDataTransfer.cpp b/mlir/lib/Transforms/PipelineDataTransfer.cpp index 0d025f5678f..87899564172 100644 --- a/mlir/lib/Transforms/PipelineDataTransfer.cpp +++ b/mlir/lib/Transforms/PipelineDataTransfer.cpp @@ -100,7 +100,8 @@ static bool doubleBuffer(MLValue *oldMemRef, ForStmt *forStmt) { ->getResult()); auto d0 = bInner.getAffineDimExpr(0); - auto *modTwoMap = bInner.getAffineMap(1, 0, {d0 % 2}, {}); + auto modTwoMap = + bInner.getAffineMap(/*dimCount=*/1, /*symbolCount=*/0, {d0 % 2}, {}); auto ivModTwoOp = bInner.create<AffineApplyOp>(forStmt->getLoc(), modTwoMap, forStmt); if (!replaceAllMemRefUsesWith(oldMemRef, newMemRef, ivModTwoOp->getResult(0))) |
