[CODEGEN][COREML] Call InferType explicitly in coreml test (apache#6676)

TusharKanekiDey · Oct 14, 2020 · 9e7c5af · 9e7c5af
1 parent 6395eb3
commit 9e7c5af
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/tests/python/contrib/test_coreml_codegen.py b/tests/python/contrib/test_coreml_codegen.py
@@ -76,12 +76,14 @@ def _create_graph_annotated():
     func2 = func2.with_attr("global_symbol", target + "_2")
     gv2 = relay.GlobalVar(target + "_2")
     mod[gv2] = func2
+    mod = relay.transform.InferType()(mod)
 
     # body
     x = relay.var("x", shape=shape)
     y = relay.var("y", shape=shape)
     func = relay.Function([x, y], gv0(y) - gv2(x))
     mod["main"] = func
+    mod = relay.transform.InferType()(mod)
 
     return mod