OSchip
/
llvm-project
mirror of https://github.com/THU-DSP-LAB/llvm-project.git
13
0
Fork 2
Code Issues Packages Projects Releases Wiki Activity

Add Builder::getI64VectorAttr. 

This matches Builder::getI32VectorAttr.

Differential Revision: https://reviews.llvm.org/D75883
This commit is contained in:
Sean Silva 2020-03-11 15:15:06 -07:00
parent 18fc42fa33
commit 94f848d7b5
2 changed files with 10 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
mlir/include/mlir/IR/Builders.h
View File

@ -117,6 +117,7 @@ public:
IntegerAttr getUI32IntegerAttr(uint32_t value);
DenseIntElementsAttr getI32VectorAttr(ArrayRef<int32_t> values);
DenseIntElementsAttr getI64VectorAttr(ArrayRef<int64_t> values);
ArrayAttr getAffineMapArrayAttr(ArrayRef<AffineMap> values);
ArrayAttr getI32ArrayAttr(ArrayRef<int32_t> values);

14
mlir/lib/IR/Builders.cpp
View File

@ -98,11 +98,15 @@ IntegerAttr Builder::getI64IntegerAttr(int64_t value) {
}
DenseIntElementsAttr Builder::getI32VectorAttr(ArrayRef<int32_t> values) {
return DenseElementsAttr::get(
VectorType::get(static_cast<int64_t>(values.size()),
getIntegerType(32)),
values)
.cast<DenseIntElementsAttr>();
return DenseIntElementsAttr::get(
VectorType::get(static_cast<int64_t>(values.size()), getIntegerType(32)),
values);
}
DenseIntElementsAttr Builder::getI64VectorAttr(ArrayRef<int64_t> values) {
return DenseIntElementsAttr::get(
VectorType::get(static_cast<int64_t>(values.size()), getIntegerType(64)),
values);
}
IntegerAttr Builder::getI32IntegerAttr(int32_t value) {