[mlir] Use getDenseElementBitwidth instead of Type::getElementTypeBitWidth.

[rriddle]

Summary

Some data values have a different storage width than the corresponding MLIR type, e.g. bfloat is currently stored as a double.

[merge_guards_bot]

Unit tests: pass. 61657 tests passed, 0 failed and 778 were skipped.

clang-tidy: fail. Please fix clang-tidy findings.

clang-format: pass.

Build artifacts: diff.json, clang-tidy.txt, clang-format.patch, CMakeCache.txt, console-log.txt, test-results.xml

mlir/lib/IR/Attributes.cpp

 }
 
 /// Check the information for a c++ data type, check if this type is valid for
 /// the current attribute. This method is used to verify specific type
 /// invariants that the templatized 'getValues' method cannot.
 static bool isValidIntOrFloat(ShapedType type, int64_t dataEltSize,
                               bool isInt) {
   // Make sure that the data element size is the same as the type element width.
-  if ((dataEltSize * CHAR_BIT) != type.getElementTypeBitWidth())
+  if (getDenseElementBitwidth(type.getElementType()) !=
+      static_cast<size_t>(dataEltSize * CHAR_BIT))
     return false;
 
   // Check that the element type is valid.
   return isInt ? type.getElementType().isa<IntegerType>()
                : type.getElementType().isa<FloatType>();
 }
 
 /// Overload of the 'getRaw' method that asserts that the given type is of

mlir/unittests/IR/AttributeTest.cpp

 
 TEST(DenseSplatTest, FloatAttrSplat) {
   MLIRContext context;
   FloatType floatTy = FloatType::getBF16(&context);
   Attribute value = FloatAttr::get(floatTy, 10.0);
 
   testSplat(floatTy, value);
 }
+
+TEST(DenseSplatTest, BF16Splat) {
+  MLIRContext context;
+  FloatType floatTy = FloatType::getBF16(&context);
+  // Note: We currently use double to represent bfloat16.
+  double value = 10.0;
+
+  testSplat(floatTy, value);
+}
+
 } // end namespace