fix build

apache · Feb 7, 2019 · d685e71 · d685e71
1 parent 3d363db
commit d685e71
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/src/relay/pass/type_infer.cc b/src/relay/pass/type_infer.cc
@@ -438,14 +438,14 @@ class TypeInferencer : private ExprFunctor<Type(const Expr&)> {
 
   Type VisitExpr_(const RefReadNode* op) final {
     Type it = IncompleteTypeNode::make(TypeVarNode::Kind::kType);
-    this->Unify(GetType(op->ref), RefTypeNode::make(it), op->span);
+    this->Unify(GetType(op->ref), RefTypeNode::make(it), GetRef<RefRead>(op));
     return it;
   }
 
   Type VisitExpr_(const RefWriteNode* op) final {
     Type it = IncompleteTypeNode::make(TypeVarNode::Kind::kType);
-    this->Unify(GetType(op->ref), RefTypeNode::make(it), op->span);
-    this->Unify(GetType(op->value), it, op->span);
+    this->Unify(GetType(op->ref), RefTypeNode::make(it), GetRef<RefWrite>(op));
+    this->Unify(GetType(op->value), it, GetRef<RefWrite>(op));
     return TupleTypeNode::make({});
   }
 };