Add ArrayAttr for memory space representation (llvm#45)

hrotuna · Jul 25, 2022 · 2f4572a · 2f4572a
1 parent b66e2f6
commit 2f4572a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/mlir/lib/IR/BuiltinTypes.cpp b/mlir/lib/IR/BuiltinTypes.cpp
@@ -472,7 +472,7 @@ bool mlir::detail::isSupportedMemorySpace(Attribute memorySpace) {
     return true;
 
   // Supported built-in attributes.
-  if (memorySpace.isa<IntegerAttr, StringAttr, DictionaryAttr, SymbolRefAttr>())
+  if (memorySpace.isa<IntegerAttr, StringAttr, DictionaryAttr, SymbolRefAttr, ArrayAttr>())
     return true;
 
   // Allow custom dialect attributes.