save

apache · Jun 28, 2019 · 837be95 · 837be95
1 parent cc02471
commit 837be95
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/src/relay/ir/type_functor.cc b/src/relay/ir/type_functor.cc
@@ -221,7 +221,7 @@ class TypeBinder : public TypeMutator {
 };
 
 Type Bind(const Type& type, const tvm::Map<TypeVar, Type>& args_map) {
-  return TypeBinder(args_map).VisitType(type);
+  return type.defined() ? TypeBinder(args_map).VisitType(type) : type;
 }
 
 }  // namespace relay

diff --git a/tests/python/relay/test_pass_to_cps.py b/tests/python/relay/test_pass_to_cps.py
@@ -69,6 +69,7 @@ def test_ad_cps_pe():
     print(F)
     relay.ir_pass.infer_type(F)
     F = relay.ir_pass.partial_evaluate(F)
+    F = relay.ir_pass.dead_code_elimination(F)
     print(F)
 
 if __name__ == '__main__':