Add convenience builder for MemRefType.
PiperOrigin-RevId: 208245701
diff --git a/lib/IR/Builders.cpp b/lib/IR/Builders.cpp
index 110e440..be334ef 100644
--- a/lib/IR/Builders.cpp
+++ b/lib/IR/Builders.cpp
@@ -59,6 +59,12 @@
return FunctionType::get(inputs, results, context);
}
+MemRefType *Builder::getMemRefType(ArrayRef<int> shape, Type *elementType,
+ ArrayRef<AffineMap *> affineMapComposition,
+ unsigned memorySpace) {
+ return MemRefType::get(shape, elementType, affineMapComposition, memorySpace);
+}
+
VectorType *Builder::getVectorType(ArrayRef<unsigned> shape,
Type *elementType) {
return VectorType::get(shape, elementType);