commit b61885d605c6dbaf94b3bb535bdc90cc0199ab7b
author MLIR Team <no-reply@google.com> Wed Jul 18 16:29:21 2018 -0700
committer jpienaar <jpienaar@google.com> Fri Mar 29 12:35:40 2019 -0700
tree 35b73e9f984656450181a91ef09c489c2420fc5d
parent 3b2ef76422a5d72ae28c0c6cba5b733a0402df4d

Support for AffineMapAttr.

PiperOrigin-RevId: 205157390

lib/Parser/Parser.cpp

diff --git a/lib/Parser/Parser.cpp b/lib/Parser/Parser.cpp
index 3d5c908..d220787 100644
--- a/lib/Parser/Parser.cpp
+++ b/lib/Parser/Parser.cpp
@@ -598,6 +598,11 @@
     return builder.getArrayAttr(elements);
   }
   default:
+    // Try to parse affine map reference.
+    auto* affineMap = parseAffineMapReference();
+    if (affineMap != nullptr)
+      return builder.getAffineMapAttr(affineMap);
+
     // TODO: Handle floating point.
     return (emitError("expected constant attribute value"), nullptr);
   }