From d29b208eb826ad9827951c2d0c4cb8a4a646ca06 Mon Sep 17 00:00:00 2001
From: MORITA Kazutaka <morita.kazutaka@gmail.com>
Date: Tue, 13 Oct 2020 11:34:50 +0900
Subject: [PATCH] [CODEGEN][COREML] Call InferType explicitly in coreml test

---
 tests/python/contrib/test_coreml_codegen.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/tests/python/contrib/test_coreml_codegen.py b/tests/python/contrib/test_coreml_codegen.py
index 018aadeee4eb..dd10b6d9fcbd 100644
--- 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