diff --git a/src/relay/ir/pretty_printer.cc b/src/relay/ir/pretty_printer.cc index f4a830040f70..71502614abd6 100644 --- a/src/relay/ir/pretty_printer.cc +++ b/src/relay/ir/pretty_printer.cc @@ -592,6 +592,22 @@ class PrettyPrinter : return AllocTypeVar(GetRef(node)); } + Doc VisitType_(const GlobalTypeVarNode* node) final { + return Doc(node->var->name_hint); + } + + Doc VisitType_(const TypeCallNode* node) final { + Doc doc = PrintType(node->func, false); + std::vector args; + for (const Type& t : node->args) { + args.push_back(PrintType(t, false)); + } + doc << "["; + doc << PrintVec(args); + doc << "]"; + return doc; + } + Doc VisitType_(const TensorTypeNode* node) final { // scalar type if (node->shape.size() == 0) {