commit c03c695772f409f9ee0a55229983c98cb77d5403
author Jacques Pienaar <jpienaar@google.com> Fri Aug 10 11:56:47 2018 -0700
committer jpienaar <jpienaar@google.com> Fri Mar 29 12:58:32 2019 -0700
tree 0518ab526a2f7c97dd5daeba14ea3b104d245eb5
parent b697ac137ee95284e76c6198405a3cf775ece4cf

Add convenience builder for MemRefType.

PiperOrigin-RevId: 208245701

include/mlir/IR/Builders.h

diff --git a/include/mlir/IR/Builders.h b/include/mlir/IR/Builders.h
index af089e6..3324d4b 100644
--- a/include/mlir/IR/Builders.h
+++ b/include/mlir/IR/Builders.h
@@ -70,6 +70,9 @@
   IntegerType *getIntegerType(unsigned width);
   FunctionType *getFunctionType(ArrayRef<Type *> inputs,
                                 ArrayRef<Type *> results);
+  MemRefType *getMemRefType(ArrayRef<int> shape, Type *elementType,
+                            ArrayRef<AffineMap *> affineMapComposition = {},
+                            unsigned memorySpace = 0);
   VectorType *getVectorType(ArrayRef<unsigned> shape, Type *elementType);
   RankedTensorType *getTensorType(ArrayRef<int> shape, Type *elementType);
   UnrankedTensorType *getTensorType(Type *elementType);