huajsj commented on code in PR #11334: URL: https://github.com/apache/tvm/pull/11334#discussion_r875528806
########## tests/python/relay/test_pipeline_executor.py: ########## @@ -22,12 +22,193 @@ import tvm import tvm.testing from tvm import relay -from tvm.relay import transform +from tvm.relay import transform, build_module +from tvm.relay.testing import run_opt_pass from tvm.contrib import graph_executor, pipeline_executor, pipeline_executor_build from tvm._ffi import get_global_func from tvm.contrib import cc as _cc +"""Splitting graph into a list of subgraph""" + + +def graph_split(expr, split_conf, params=None): + def get_dep_var(sub_var_dep): + return [var for var, _ in sub_var_dep[len(sub_var_dep) - 1]["ref_nodes"].items()] Review Comment: the function need to get the key only so we can not remove the ',_' . -- 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]
