diff options
Diffstat (limited to 'mlir/lib/IR/Builders.cpp')
| -rw-r--r-- | mlir/lib/IR/Builders.cpp | 8 |
1 files changed, 8 insertions, 0 deletions
diff --git a/mlir/lib/IR/Builders.cpp b/mlir/lib/IR/Builders.cpp index afdeefd023c..4d6cd3550ca 100644 --- a/mlir/lib/IR/Builders.cpp +++ b/mlir/lib/IR/Builders.cpp @@ -100,6 +100,14 @@ IntegerAttr Builder::getI64IntegerAttr(int64_t value) { return IntegerAttr::get(getIntegerType(64), APInt(64, value)); } +DenseIntElementsAttr Builder::getI32VectorAttr(ArrayRef<int32_t> values) { + return DenseElementsAttr::get( + VectorType::get(static_cast<int64_t>(values.size()), + getIntegerType(32)), + values) + .cast<DenseIntElementsAttr>(); +} + IntegerAttr Builder::getI32IntegerAttr(int32_t value) { return IntegerAttr::get(getIntegerType(32), APInt(32, value)); } |

