Do not attach checked types to constructors

apache · Jan 31, 2019 · a4c1907 · a4c1907
1 parent 6a5a9e3
commit a4c1907
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/relay/pass/type_infer.cc b/src/relay/pass/type_infer.cc
@@ -518,7 +518,7 @@ class TypeInferencer::Resolver : public ExprMutator, PatternMutator {
   }
 
   Expr VisitExpr_(const ConstructorNode* op) final {
-    return AttachCheckedType(op);
+    return GetRef<Constructor>(op);
   }
 
   Expr VisitExpr_(const MatchNode* op) final {
@@ -657,6 +657,7 @@ struct AllCheckTypePopulated : ExprVisitor {
   void VisitExpr(const Expr& e) {
     if (e.as<OpNode>()) { return; }
     if (e.as<GlobalVarNode>()) { return; }
+    if (e.as<ConstructorNode>()) { return; }
     CHECK(e->checked_type_.defined()) << "Expression: " << e;
     return ExprVisitor::VisitExpr(e);
   }