Mousius commented on a change in pull request #8280:
URL: https://github.com/apache/tvm/pull/8280#discussion_r674765137
##########
File path: python/tvm/micro/model_library_format.py
##########
@@ -207,6 +208,42 @@ def _build_function_memory_map(function_metadata):
return ret
+def _get_main_relay_func(mod: executor_factory.ExecutorFactoryModule):
+ main_func = mod.function_metadata[MAIN_FUNC_NAME_STR]
+ target = list(main_func.relay_primfuncs.keys())[0]
+ return main_func.relay_primfuncs[target]
+
+
+def _convert_tuple_to_outputs(ret_type, offset=0):
+ outputs = []
+ added_fields = len(ret_type.fields)
+ for output_index in range(added_fields):
+ next_output = offset + len(outputs)
+ if isinstance(ret_type.fields[output_index], TupleType):
+
outputs.extend(_convert_tuple_to_outputs(ret_type.fields[output_index],
next_output))
+ else:
+ outputs.append(f"output{next_output}")
+ return outputs
+
+
+def _get_inputs_and_outputs_from_module(mod):
+ main_func = _get_main_relay_func(mod)
+ inputs = [argument.name_hint for argument in main_func.params]
+
+ outputs = ["output"]
+ if isinstance(main_func.ret_type, TupleType):
+ outputs = _convert_tuple_to_outputs(main_func.ret_type)
+
+ return inputs, outputs
+
+
+def _should_generate_interface_header(mod):
+ for _, target in mod.target.items():
Review comment:
Fantastic, I'll give it a go :smile_cat: !
--
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]
