Enhance the type system to support arbitrary precision integers, which are

important for low-bitwidth inference cases and hardware synthesis targets.

Rename 'int' to 'affineint' to avoid confusion between "the integers" and "the int
type".

PiperOrigin-RevId: 202751508

1 files changed, 12 insertions, 0 deletions

diff --git a/mlir/lib/Parser/Token.cpp b/mlir/lib/Parser/Token.cpp
index 5563255b444..e1e4bedf28f 100644
--- a/mlir/lib/Parser/Token.cpp
+++ b/mlir/lib/Parser/Token.cpp
@@ -48,6 +48,18 @@ Optional<unsigned> Token::getUnsignedIntegerValue() const {
   return result;
 }
 
+/// For an inttype token, return its bitwidth.
+Optional<unsigned> Token::getIntTypeBitwidth() const {
+ unsigned result = 0;
+  if (spelling[1] == '0' ||
+      spelling.drop_front().getAsInteger(10, result) ||
+      // Arbitrary but large limit on bitwidth.
+      result > 4096 || result == 0)
+    return None;
+  return result;
+}
+
+
 /// Given a 'string' token, return its value, including removing the quote
 /// characters and unescaping the contents of the string.
 std::string Token::getStringValue() const {