From 723d3b836a81ebad238a5abc5041cd8b80b61c67 Mon Sep 17 00:00:00 2001 From: Manupa Karunaratne Date: Mon, 25 Apr 2022 11:44:04 +0100 Subject: [PATCH] fixup! [USMP] Adding support for U4 usecase Change-Id: I857f3d0ba7bc192d56d750c44b232998b2876e7a --- src/relay/backend/aot_executor_codegen.cc | 9 +++++++-- 1 file changed, 7 insertions(+), 2 deletions(-) diff --git a/src/relay/backend/aot_executor_codegen.cc b/src/relay/backend/aot_executor_codegen.cc index 2b3a6d7dde78..22d4b1c032f4 100644 --- a/src/relay/backend/aot_executor_codegen.cc +++ b/src/relay/backend/aot_executor_codegen.cc @@ -1081,10 +1081,15 @@ class AOTExecutorCodegen : public MixedModeVisitor { if (lowered_main_func->body->IsInstance()) { Tuple output_tuple = Downcast(lowered_main_func->body); for (unsigned i = 0; i < output_tuple->fields.size(); i++) { - CreateIOVar(output_tuple->fields[i], output_tensor_names[i]); + // AoT Executor Codegen does not create these names, + // thus should be used as they are provided. + CreateIOVar(output_tuple->fields[i], output_tensor_names[i], + /*use_unique_name = */ false); } } else { - CreateIOVar(lowered_main_func->body, output_tensor_names[0]); + // AoT Executor Codegen does not create these names, + // thus should be used as they are provided. + CreateIOVar(lowered_main_func->body, output_tensor_names[0], /*use_unique_name = */ false); } } else { // If output tensor names are not provided we will generate output(x)