Check for Vector or Tensor Type rather than ShapedType when matching constant...

    Only vectors or tensors can be constant splats (or element attrs). This is in preparation for making MemRef subclass ShapedType

--

PiperOrigin-RevId: 250586281

diff --git a/mlir/include/mlir/IR/Matchers.h b/mlir/include/mlir/IR/Matchers.h
index d1c2f94..61796ff 100644 (file)
--- a/mlir/include/mlir/IR/Matchers.h
+++ b/mlir/include/mlir/IR/Matchers.h
@@ -101,7 +101,7 @@ struct constant_int_op_binder {
     if (type.isa<IntegerType>()) {
       return attr_value_binder<IntegerAttr>(bind_value).match(attr);
     }
-    if (type.isa<ShapedType>()) {
+    if (type.isa<VectorType>() || type.isa<RankedTensorType>()) {
       if (auto splatAttr = attr.dyn_cast<SplatElementsAttr>()) {
         return attr_value_binder<IntegerAttr>(bind_value)
             .match(splatAttr.getValue());