This is an automated email from the ASF dual-hosted git repository.
zhic pushed a commit to branch main
in repository https://gitbox.apache.org/repos/asf/incubator-tvm.git
The following commit(s) were added to refs/heads/main by this push:
new f196a81 [CODEGEN][COREML] Call InferType explicitly in coreml test
(#6676)
f196a81 is described below
commit f196a81e73322776ca1c70d413cc266bacaa42cc
Author: MORITA Kazutaka <[email protected]>
AuthorDate: Wed Oct 14 00:54:06 2020 +0900
[CODEGEN][COREML] Call InferType explicitly in coreml test (#6676)
---
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 018aade..dd10b6d 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
