diff options
Diffstat (limited to 'mlir/lib/Transforms/MaterializeVectors.cpp')
| -rw-r--r-- | mlir/lib/Transforms/MaterializeVectors.cpp | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/mlir/lib/Transforms/MaterializeVectors.cpp b/mlir/lib/Transforms/MaterializeVectors.cpp index 6085edd8e8e..e82390f8db9 100644 --- a/mlir/lib/Transforms/MaterializeVectors.cpp +++ b/mlir/lib/Transforms/MaterializeVectors.cpp @@ -187,7 +187,7 @@ struct MaterializationState { MaterializationState() : hwVectorSize(clVectorSize.size(), 0) { std::copy(clVectorSize.begin(), clVectorSize.end(), hwVectorSize.begin()); } - SmallVector<int, 8> hwVectorSize; + SmallVector<int64_t, 8> hwVectorSize; VectorType superVectorType; VectorType hwVectorType; SmallVector<unsigned, 8> hwVectorInstance; @@ -458,7 +458,7 @@ static AffineMap projectedPermutationMap(VectorTransferOpTy *transfer, SmallVector<AffineExpr, 4> keep; MLIRContext *context = transfer->getInstruction()->getContext(); functional::zipApply( - [&dim, &keep, context](int shape, int ratio) { + [&dim, &keep, context](int64_t shape, int64_t ratio) { assert(shape >= ratio && "shape dim must be greater than ratio dim"); if (shape != ratio) { // HW vector is not full instantiated along this dim, keep it. |
