yongwww opened a new pull request, #14176: URL: https://github.com/apache/tvm/pull/14176
I am trying to add return support from true/false branch of `IfNode` as we discussed in https://github.com/tlc-pack/relax/issues/446. I introduced a `RelaxReturnGlobalInfo` (similar to `DummyGlobalInfo`), which is used to save all the `Expr` (`body` of `SeqExpr` ) that need to be returned. But this causes it only works at IR_Module level, it failed to work for relax function without explicitly defining in ir_module. Options I have at this moment. - o0: Leverage metadata to save all the return Exprs. - o1: Introduce GlobalInfo in relax Function level, a new member like `global_info` will be added into FunctionNode - o2: Add a new member tag `return_body` into `SeqExpr` (we don't want to change IR at this stage, list it here for consideration in the future) Any comments are truly appreciated! cc: @slyubomirsky @tqchen @junrushao @Hzfengsy @YuchenJin -- This is an automated message from the Apache Git Service. To respond to the message, please log on to GitHub and use the URL above to go to the specific comment. To unsubscribe, e-mail: [email protected] For queries about this service, please contact Infrastructure at: [email protected]
