Merge pull request #463 from facebookresearch/schedule-extension-newline

schedule_print.cc: print each map of an extension on a new line

Author: ftynse

tc/core/polyhedral/schedule_print.cc

@@ -185,7 +185,12 @@ std::ostream& ScheduleTreeElemDomain::write(std::ostream& os) const {
 
 std::ostream& ScheduleTreeElemExtension::write(std::ostream& os) const {
   WS w;
-  os << w.tab() << "extension(" << extension_ << ")";
+  os << w.tab() << "extension(";
+  for (const auto& e : extension_.get_map_list()) {
+    WS w2;
+    os << std::endl << w2.tab() << e;
+  }
+  os << ")";
   return os;
 }