From da646505c5bc5f03252c7a83bbeb890e5b3a19fa Mon Sep 17 00:00:00 2001
From: Smit Hinsu <hinsu@google.com>
Date: Mon, 2 Sep 2019 23:43:36 -0700
Subject: [PATCH] Support bf16 in Builder::getZeroAttr

PiperOrigin-RevId: 266863802
---
 mlir/lib/IR/Builders.cpp | 5 ++---
 1 file changed, 2 insertions(+), 3 deletions(-)

diff --git a/mlir/lib/IR/Builders.cpp b/mlir/lib/IR/Builders.cpp
index 067ff7a..d7b4883 100644
--- a/mlir/lib/IR/Builders.cpp
+++ b/mlir/lib/IR/Builders.cpp
@@ -253,12 +253,11 @@ ArrayAttr Builder::getAffineMapArrayAttr(ArrayRef<AffineMap> values) {
 
 Attribute Builder::getZeroAttr(Type type) {
   switch (type.getKind()) {
+  case StandardTypes::BF16:
   case StandardTypes::F16:
-    return getF16FloatAttr(0);
   case StandardTypes::F32:
-    return getF32FloatAttr(0);
   case StandardTypes::F64:
-    return getF64FloatAttr(0);
+    return getFloatAttr(type, 0.0);
   case StandardTypes::Integer: {
     auto width = type.cast<IntegerType>().getWidth();
     if (width == 1)
-- 
2.7.4