commit 0115dbb2dae18394d9477541e3dd1959c9e17778
author Uday Bondhugula <bondhugula@google.com> Wed Jul 11 21:31:07 2018 -0700
committer jpienaar <jpienaar@google.com> Fri Mar 29 12:32:59 2019 -0700
tree cd8aefbbadb6477f8a07dd7b38fc85aa88644a5b
parent e082aadfadecf4ac9a748df9bd7db01289ae8f56

Parse affine map range sizes.

PiperOrigin-RevId: 204240947

lib/IR/AsmPrinter.cpp

diff --git a/lib/IR/AsmPrinter.cpp b/lib/IR/AsmPrinter.cpp
index d284d0f..27ed3e4 100644
--- a/lib/IR/AsmPrinter.cpp
+++ b/lib/IR/AsmPrinter.cpp
@@ -392,6 +392,17 @@
   os << " -> (";
   interleave(getResults(), [&](AffineExpr *expr) { os << *expr; },
              [&]() { os << ", "; });
+  os << ")";
+
+  if (!isBounded()) {
+    os << "\n";
+    return;
+  }
+
+  // Print range sizes for bounded affine maps.
+  os << " size (";
+  interleave(getRangeSizes(), [&](AffineExpr *expr) { os << *expr; },
+             [&]() { os << ", "; });
   os << ")\n";
 }