fixup! [USMP] Adding support for U4 usecase

Change-Id: I857f3d0ba7bc192d56d750c44b232998b2876e7a
apache · Apr 25, 2022 · 723d3b8 · 723d3b8
1 parent 662d278
commit 723d3b8
Showing 1 changed file with 7 additions and 2 deletions.
diff --git 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<TupleNode>()) {
         Tuple output_tuple = Downcast<Tuple>(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)