Support FP16 in getZeroAttr.

PiperOrigin-RevId: 251783931

diff --git a/mlir/include/mlir/IR/Builders.h b/mlir/include/mlir/IR/Builders.h
index 52a78ed..c192806 100644 (file)
--- a/mlir/include/mlir/IR/Builders.h
+++ b/mlir/include/mlir/IR/Builders.h
@@ -132,6 +132,7 @@ public:
   Attribute getZeroAttr(Type type);
 
   // Convenience methods for fixed types.
+  FloatAttr getF16FloatAttr(float value);
   FloatAttr getF32FloatAttr(float value);
   FloatAttr getF64FloatAttr(double value);
   IntegerAttr getI32IntegerAttr(int32_t value);

diff --git a/mlir/lib/IR/Builders.cpp b/mlir/lib/IR/Builders.cpp
index ac06609..63efe47 100644 (file)
--- a/mlir/lib/IR/Builders.cpp
+++ b/mlir/lib/IR/Builders.cpp
@@ -147,6 +147,10 @@ FloatAttr Builder::getF32FloatAttr(float value) {
   return FloatAttr::get(getF32Type(), APFloat(value));
 }
 
+FloatAttr Builder::getF16FloatAttr(float value) {
+  return FloatAttr::get(getF16Type(), value);
+}
+
 FloatAttr Builder::getFloatAttr(Type type, double value) {
   return FloatAttr::get(type, value);
 }
@@ -242,6 +246,8 @@ ArrayAttr Builder::getStrArrayAttr(ArrayRef<StringRef> values) {
 
 Attribute Builder::getZeroAttr(Type type) {
   switch (type.getKind()) {
+  case StandardTypes::F16:
+    return getF16FloatAttr(0);
   case StandardTypes::F32:
     return getF32FloatAttr(0);
   case StandardTypes::F64: