diff --git a/src/relay/pass/type_infer.cc b/src/relay/pass/type_infer.cc index 5d78ba2516855..8582030145572 100644 --- a/src/relay/pass/type_infer.cc +++ b/src/relay/pass/type_infer.cc @@ -345,7 +345,7 @@ class TypeInferencer : private ExprFunctor { if (inc_ty_node != nullptr) { Type ret_type = IncompleteTypeNode::make(TypeVarNode::Kind::kType); Type func_type = FuncTypeNode::make(arg_types, ret_type, {}, {}); - Type unified = this->Unify(ftype, func_type, call->span); + Type unified = this->Unify(ftype, func_type, call); fn_ty_node = unified.as(); }