From a3cfa5ff9ec1160dbc5edbb5aa5a40d9e4ac7f7d Mon Sep 17 00:00:00 2001
From: Tianqi Chen <tqchen@users.noreply.github.com>
Date: Wed, 14 Nov 2018 18:59:39 -0800
Subject: [PATCH] [RELAY] bugfix type functor caching (#2113)
---
src/relay/ir/type_functor.cc | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/src/relay/ir/type_functor.cc b/src/relay/ir/type_functor.cc
index fc0daa3cb..0ef1743cb 100644
--- a/src/relay/ir/type_functor.cc
+++ b/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,
--
GitLab