Tweak printing for now

jroesch · Mar 19, 2021 · 7ce8f66 · 7ce8f66
1 parent c103a23
commit 7ce8f66
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/src/relay/ir/function.cc b/src/relay/ir/function.cc
@@ -62,9 +62,11 @@ TVM_REGISTER_GLOBAL("relay.ir.Function")
 
 TVM_STATIC_IR_FUNCTOR(ReprPrinter, vtable)
     .set_dispatch<FunctionNode>([](const ObjectRef& ref, ReprPrinter* p) {
-      auto* node = static_cast<const FunctionNode*>(ref.get());
-      p->stream << "FunctionNode(" << node->params << ", " << node->ret_type << ", " << node->body
-                << ", " << node->type_params << ", " << node->attrs << ")";
+      // auto* node = static_cast<const FunctionNode*>(ref.get());
+      // p->stream << "FunctionNode(" << node->params << ", " << node->ret_type << ", " << node->body
+      //           << ", " << node->type_params << ", " << node->attrs << ")";
+      // we should standardize behavior
+      p->stream << PrettyPrint(ref);
     });
 
 }  // namespace relay