[GRAPHIR] Print attributes of input (#69)

eric-haibin-lin · Oct 6, 2017 · f78c9c1 · f78c9c1
1 parent 73e6274
commit f78c9c1
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/src/pass/print_graph_ir.cc b/src/pass/print_graph_ir.cc
@@ -109,6 +109,17 @@ void PrintGraphIR_(Graph src,
     }
   };
 
+  if (trigger.size() != 0) {
+    for (size_t i = 0; i < idx.input_nodes().size(); ++i) {
+      uint32_t nid = idx.input_nodes()[i];
+      os << "  %" << idx[nid].source->attrs.name;
+      for (const auto& fp : trigger) {
+        fp(nid, os);
+      }
+      os << '\n';
+    }
+  }
+
   for (uint32_t nid = 0; nid < idx.num_nodes(); ++nid) {
     const auto& inode = idx[nid];
     if (inode.source->is_variable()) continue;