From 8d40302b3403f91f25670d7136ab5ee668f11d45 Mon Sep 17 00:00:00 2001
From: Rob Suderman <suderman@google.com>
Date: Wed, 3 Apr 2019 14:30:17 -0700
Subject: [PATCH]     Added memref to OpBase.td as specific memref types should
 be supported by default.

--

PiperOrigin-RevId: 241806461
---
 mlir/include/mlir/IR/OpBase.td | 23 +++++++++++++++++++++++
 1 file changed, 23 insertions(+)

diff --git a/mlir/include/mlir/IR/OpBase.td b/mlir/include/mlir/IR/OpBase.td
index 3c2ad54..3deddd0 100644
--- a/mlir/include/mlir/IR/OpBase.td
+++ b/mlir/include/mlir/IR/OpBase.td
@@ -158,6 +158,9 @@ def IsVectorOrTensorTypePred : CPred<"{0}.isa<VectorOrTensorType>()">;
 // Whether a type is a TupleType.
 def IsTupleTypePred : CPred<"{0}.isa<TupleType>()">;
 
+// Whether a type is a MemRefType.
+def IsMemRefTypePred : CPred<"{0}.isa<MemRefType>()">;
+
 // For a TensorType, verify that it is a statically shaped tensor.
 def IsStaticShapeTensorTypePred :
   CPred<"{0}.cast<TensorType>().hasStaticShape()">;
@@ -318,6 +321,26 @@ def F64Tensor  : TypedTensor<F64>;
 // there is not only a single elemental type.
 def Tuple : Type<IsTupleTypePred, "tuple">;
 
+// Memref type.
+
+// Memrefs are blocks of data with fixed type and rank.
+class MemRef<Type t>
+    : ContainerType<t, IsMemRefTypePred,
+                    "{0}.cast<MemRefType>().getElementType()", "memref">;
+
+// Memref declarations handle any memref, independent of rank, size, (static or
+// dynamic), layout, or memory space.
+def I1MemRef  : MemRef<I1>;
+def I8MemRef  : MemRef<I8>;
+def I16MemRef : MemRef<I16>;
+def I32MemRef : MemRef<I32>;
+def I64MemRef : MemRef<I64>;
+
+def BF16MemRef : MemRef<BF16>;
+def F16MemRef  : MemRef<F16>;
+def F32MemRef  : MemRef<F32>;
+def F64MemRef  : MemRef<F64>;
+
 //===----------------------------------------------------------------------===//
 // Common type constraints
 //===----------------------------------------------------------------------===//
-- 
2.7.4