squiid/llvm
1
0
Fork 0
Code Issues Pull requests Activity

Add convenience methods to create i8 and i16 attributes in Builder.

Browse source 
PiperOrigin-RevId: 269120226
This commit is contained in:
Jing Pu 2019-09-14 17:02:06 -07:00 committed by A. Unique TensorFlower
parent 4f32ae61b4
commit 38e7226606
2 changed files with 10 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
mlir/include/mlir/IR/Builders.h
View file

@ -131,6 +131,8 @@ public:
FloatAttr getF32FloatAttr(float value);
FloatAttr getF64FloatAttr(double value);
IntegerAttr getI8IntegerAttr(int8_t value);
IntegerAttr getI16IntegerAttr(int16_t value);
IntegerAttr getI32IntegerAttr(int32_t value);
IntegerAttr getI64IntegerAttr(int64_t value);

8
mlir/lib/IR/Builders.cpp
View file

@ -123,6 +123,14 @@ IntegerAttr Builder::getI32IntegerAttr(int32_t value) {
return IntegerAttr::get(getIntegerType(32), APInt(32, value));
}
IntegerAttr Builder::getI16IntegerAttr(int16_t value) {
return IntegerAttr::get(getIntegerType(16), APInt(16, value));
}
IntegerAttr Builder::getI8IntegerAttr(int8_t value) {
return IntegerAttr::get(getIntegerType(8), APInt(8, value));
}
IntegerAttr Builder::getIntegerAttr(Type type, int64_t value) {
if (type.isIndex())
return IntegerAttr::get(type, APInt(64, value));