Refactor SplatElementsAttr to inherit from DenseElementsAttr as opposed to being a separate Attribute type. DenseElementsAttr provides a better internal representation for splat values as well as better API for accessing elements.

PiperOrigin-RevId: 253138287

1 files changed, 2 insertions, 2 deletions

diff --git a/mlir/lib/Dialect/FxpMathOps/Transforms/UniformKernelUtils.h b/mlir/lib/Dialect/FxpMathOps/Transforms/UniformKernelUtils.h
index 26428cfb409..f0eeba0891a 100644
--- a/mlir/lib/Dialect/FxpMathOps/Transforms/UniformKernelUtils.h
+++ b/mlir/lib/Dialect/FxpMathOps/Transforms/UniformKernelUtils.h
@@ -190,7 +190,7 @@ inline Type castElementType(Type t, Type newElementType) {
 inline Attribute broadcastScalarConstIntValue(Type t, int64_t value) {
   if (auto st = t.dyn_cast<ShapedType>()) {
     assert(st.getElementType().isa<IntegerType>());
-    return SplatElementsAttr::get(st,
+    return DenseElementsAttr::get(st,
                                   IntegerAttr::get(st.getElementType(), value));
   }
 
@@ -219,7 +219,7 @@ inline Attribute broadcastScalarConstFloatValue(Type t, APFloat value) {
     assert(floatElementType &&
            "float broadcast element type must be float like");
     APFloat apValue = convertFloatToType(floatElementType, value);
-    return SplatElementsAttr::get(st,
+    return DenseElementsAttr::get(st,
                                   FloatAttr::get(st.getElementType(), apValue));
   } else {
     auto floatType = t.dyn_cast<FloatType>();