Skip to content

Commit f196a81

Browse files
authored
[CODEGEN][COREML] Call InferType explicitly in coreml test (#6676)
1 parent d24634a commit f196a81

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

tests/python/contrib/test_coreml_codegen.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -76,12 +76,14 @@ def _create_graph_annotated():
7676
func2 = func2.with_attr("global_symbol", target + "_2")
7777
gv2 = relay.GlobalVar(target + "_2")
7878
mod[gv2] = func2
79+
mod = relay.transform.InferType()(mod)
7980

8081
# body
8182
x = relay.var("x", shape=shape)
8283
y = relay.var("y", shape=shape)
8384
func = relay.Function([x, y], gv0(y) - gv2(x))
8485
mod["main"] = func
86+
mod = relay.transform.InferType()(mod)
8587

8688
return mod
8789

0 commit comments

Comments
 (0)