Tweak metadata variable description and MLF target loop

apache · Jul 22, 2021 · 766a874 · 766a874
1 parent c6104b8
commit 766a874
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 6 deletions.
diff --git a/include/tvm/runtime/module.h b/include/tvm/runtime/module.h
@@ -230,9 +230,9 @@ constexpr const char* tvm_module_main = "__tvm_main__";
 constexpr const char* tvm_param_prefix = "__tvm_param__";
 /*! \brief A PackedFunc that looks up linked parameters by storage_id. */
 constexpr const char* tvm_lookup_linked_param = "_lookup_linked_param";
-/*! \brief The main AOT executor function */
+/*! \brief The main AOT executor function generated from TIR */
 constexpr const char* tvm_run_func_suffix = "run_model";
-/*! \brief The models entrypoint function which calls the executor */
+/*! \brief Model entrypoint generated as an interface to the AOT function outside of TIR */
 constexpr const char* tvm_entrypoint_suffix = "run";
 }  // namespace symbol
 

diff --git a/python/tvm/micro/model_library_format.py b/python/tvm/micro/model_library_format.py
@@ -244,10 +244,7 @@ def _get_inputs_and_outputs_from_module(mod):
 
 
 def _should_generate_interface_header(mod):
-    for _, target in mod.target.items():
-        if "interface-api" in target.attrs and target.attrs["interface-api"] == "c":
-            return True
-    return False
+    return any(target.attrs.get("interface-api") == "c" for target in mod.target.values())
 
 
 def _make_tar(source_dir, tar_file_path):