From d00168ffbf290d64ee6771a59a907719469091fd Mon Sep 17 00:00:00 2001 From: Mehrdad Hessar Date: Mon, 9 Jan 2023 17:10:04 -0800 Subject: [PATCH] [CI] Fix MLF input and output name map (#13740) This PR fixes error introduced by #13704 --- python/tvm/micro/model_library_format.py | 10 ++-------- 1 file changed, 2 insertions(+), 8 deletions(-) diff --git a/python/tvm/micro/model_library_format.py b/python/tvm/micro/model_library_format.py index 11722470cd52..0f30c39ad476 100644 --- a/python/tvm/micro/model_library_format.py +++ b/python/tvm/micro/model_library_format.py @@ -468,12 +468,6 @@ def _export_graph_model_library_format( input_names = list(inputs.keys()) output_names = list(outputs.keys()) - input_name_to_size_map = { - name: property_map["size"] for name, property_map in inputs_sizes.items() - } - output_name_to_size_map = { - name: property_map["size"] for name, property_map in output_sizes.items() - } generate_c_interface_header( mod.libmod_name, input_names, @@ -483,8 +477,8 @@ def _export_graph_model_library_format( devices, workspace_size, include_path, - input_name_to_size_map, - output_name_to_size_map, + inputs_sizes, + output_sizes, ) is_aot = isinstance(mod, executor_factory.AOTExecutorFactoryModule)