diff options
| author | River Riddle <riverriddle@google.com> | 2019-01-23 14:39:45 -0800 |
|---|---|---|
| committer | jpienaar <jpienaar@google.com> | 2019-03-29 15:33:20 -0700 |
| commit | 6859f33292a7c4d908f211b9d9d8be200276d383 (patch) | |
| tree | f5a0bce4477fa4e0816d0ba0f8b68f524c1843b5 /mlir/lib/Transforms/PipelineDataTransfer.cpp | |
| parent | b64998a6b3a8d87bcd7307aa8ba8ea5008d515e4 (diff) | |
| download | bcm5719-llvm-6859f33292a7c4d908f211b9d9d8be200276d383.tar.gz bcm5719-llvm-6859f33292a7c4d908f211b9d9d8be200276d383.zip | |
Migrate VectorOrTensorType/MemRefType shape api to use int64_t instead of int.
PiperOrigin-RevId: 230605756
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 989af0071d7..9e7c928070f 100644 --- a/mlir/lib/Transforms/PipelineDataTransfer.cpp +++ b/mlir/lib/Transforms/PipelineDataTransfer.cpp @@ -87,8 +87,8 @@ static bool doubleBuffer(Value *oldMemRef, ForInst *forInst) { // Doubles the shape with a leading dimension extent of 2. auto doubleShape = [&](MemRefType oldMemRefType) -> MemRefType { // Add the leading dimension in the shape for the double buffer. - ArrayRef<int> oldShape = oldMemRefType.getShape(); - SmallVector<int, 4> newShape(1 + oldMemRefType.getRank()); + ArrayRef<int64_t> oldShape = oldMemRefType.getShape(); + SmallVector<int64_t, 4> newShape(1 + oldMemRefType.getRank()); newShape[0] = 2; std::copy(oldShape.begin(), oldShape.end(), newShape.begin() + 1); auto newMemRefType = |
