diff options
Diffstat (limited to 'mlir/lib/Conversion/VectorToLLVM/ConvertVectorToLLVM.cpp')
| -rw-r--r-- | mlir/lib/Conversion/VectorToLLVM/ConvertVectorToLLVM.cpp | 8 |
1 files changed, 4 insertions, 4 deletions
diff --git a/mlir/lib/Conversion/VectorToLLVM/ConvertVectorToLLVM.cpp b/mlir/lib/Conversion/VectorToLLVM/ConvertVectorToLLVM.cpp index 416cb4c99a3..9ec8ec6f88d 100644 --- a/mlir/lib/Conversion/VectorToLLVM/ConvertVectorToLLVM.cpp +++ b/mlir/lib/Conversion/VectorToLLVM/ConvertVectorToLLVM.cpp @@ -370,8 +370,8 @@ public: // Remaining extraction of element from 1-D LLVM vector auto position = positionAttrs.back().cast<IntegerAttr>(); - auto i32Type = LLVM::LLVMType::getInt32Ty(lowering.getDialect()); - auto constant = rewriter.create<LLVM::ConstantOp>(loc, i32Type, position); + auto i64Type = LLVM::LLVMType::getInt64Ty(lowering.getDialect()); + auto constant = rewriter.create<LLVM::ConstantOp>(loc, i64Type, position); extracted = rewriter.create<LLVM::ExtractElementOp>(loc, extracted, constant); rewriter.replaceOp(op, extracted); @@ -452,8 +452,8 @@ public: } // Insertion of an element into a 1-D LLVM vector. - auto i32Type = LLVM::LLVMType::getInt32Ty(lowering.getDialect()); - auto constant = rewriter.create<LLVM::ConstantOp>(loc, i32Type, position); + auto i64Type = LLVM::LLVMType::getInt64Ty(lowering.getDialect()); + auto constant = rewriter.create<LLVM::ConstantOp>(loc, i64Type, position); Value *inserted = rewriter.create<LLVM::InsertElementOp>( loc, lowering.convertType(oneDVectorType), extracted, adaptor.source(), constant); |
