Skip to content

T
tic
Unverified Commit a3cfa5ff by Tianqi Chen Committed by GitHub
Options

[RELAY] bugfix type functor caching (#2113)

parent a3968975
Showing with 2 additions and 2 deletions
src/relay/ir/type_functor.cc
......@@ -98,10 +98,10 @@ Type TypeMutator::VisitType_(const FuncTypeNode* op) {
}
Array<Type> new_args = MutateArray(op->arg_types);
changed = changed || new_args.same_as(op->arg_types);
changed = changed || !new_args.same_as(op->arg_types);
Type new_ret_type = VisitType(op->ret_type);
changed = changed || new_ret_type.same_as(op->ret_type);
changed = changed || !new_ret_type.same_as(op->ret_type);
if (!changed) return GetRef<Type>(op);
return FuncTypeNode::make(new_args,
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment