Add FlatSymbolRefAttr and ArrayAttr for memory space representation (#44/#45)

Author: nikita-kud

mlir/lib/IR/BuiltinTypes.cpp

@@ -474,7 +474,7 @@ bool mlir::detail::isSupportedMemorySpace(Attribute memorySpace) {
     return true;
 
   // Supported built-in attributes.
-  if (llvm::isa<IntegerAttr, StringAttr, DictionaryAttr>(memorySpace))
+  if (llvm::isa<IntegerAttr, StringAttr, DictionaryAttr, SymbolRefAttr, ArrayAttr>(memorySpace))
     return true;
 
   // Allow custom dialect attributes.