Fix pretty printer overflow

apache · Jun 9, 2020 · c45b9de · c45b9de
1 parent ed70193
commit c45b9de
Showing 1 changed file with 11 additions and 3 deletions.
diff --git a/src/printer/relay_text_printer.cc b/src/printer/relay_text_printer.cc
@@ -365,11 +365,19 @@ Doc RelayTextPrinter::VisitExpr_(const IfNode* op) {
 
 Doc RelayTextPrinter::VisitExpr_(const LetNode* op) {
   Doc doc;
-  doc << "let " << AllocVar(op->var) << " = " << Print(op->value, false, true) << ";"
-      << Doc::NewLine();
+  Expr let = GetRef<Let>(op);
+  while (auto let_node = let.as<LetNode>()) {
+    doc << "let "
+        << AllocVar(let_node->var)
+        << " = "
+        << Print(let_node->value, false, true)
+        << ";"
+        << Doc::NewLine();
+    let = let_node->body;
+  }
   // we use a scope here so GNF hoisting doesn't escape too far
   // and nested, unique lets are not hoisted
-  doc << PrintScope(op->body);
+  doc << PrintScope(let);
   return doc;
 }