[Unity][TVMScript] Fix prim_func lost issue in relax.emit_te

[yongwww]

Previously R.emit_te was introduced in #14123. The prim_funcs were not added into the same ir_module. The pr is to fix this issue, move some call_tir input handling code from bb.call_te to utils, then it is able to be leveraged by both bb.emit_te and R.emit_te.

cc: @psrivas2 @tqchen

[tvm-bot]

Thanks for contributing to TVM! Please refer to the contributing guidelines https://tvm.apache.org/docs/contribute/ for useful information and tips. Please request code reviews from Reviewers by @-ing them in a comment.

• cc @Hzfengsy, @junrushao, @quic-sanirudh, @shingjan _{See #10317 for details}

_{Generated by tvm-bot}

[tqchen]

please run the following commend to update to latest change

git rebase --onto upstream/unity upstream/unity-rebase-backup-2023-03-05 

[yongwww]

Have fixed the comments, pls take another look. cc @Hzfengsy

[Hzfengsy]

What's the difference between these two functions and DeclFunction, DefFunction?

[yongwww]

The main difference is FindModuleFrame used in DeclFunction and DefFunction, the two guys are required to be called immediately under I.ir_module(). AddFunction and UpdateFunction don't have this restriction.

Tried to use decl_function in emit_te, ran into error ValueError: 'I.DeclFunction' must be called immediately under I.ir_module()

[Hzfengsy]

Could you enhance those two instead of writing two new ones?

Excaully ValueError: 'I.DeclFunction' must be called immediately under I.ir_module() is not necessary in this case

[yongwww]

cool, thanks for letting me know that! Have removed add/update_function then. pls take another look

[psrivas2]

Thanks @yongwww !