Dependencies

apache · Nov 12, 2024 · 8b74943 · 8b74943
1 parent 74ff9bb
commit 8b74943
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 2 deletions.
diff --git a/src/runtime/crt/aot_executor_module/aot_executor_module.c b/src/runtime/crt/aot_executor_module/aot_executor_module.c
@@ -210,6 +210,8 @@ static const TVMBackendPackedCFunc aot_executor_registry_funcs[] = {
     &TVMAotExecutorModule_NotImplemented,  // set_input (implemented via python wrapper)
     &TVMAotExecutorModule_NotImplemented,  // share_params (do not implement)
     &TVMAotExecutorModule_GetInputName,    // get_input_name
+    &TVMAotExecutorModule_NotImplemented,  // get_output_index
+    &TVMAotExecutorModule_NotImplemented,  // get_output_info
 };
 
 static const TVMFuncRegistry aot_executor_registry = {
@@ -223,7 +225,9 @@ static const TVMFuncRegistry aot_executor_registry = {
     "run\0"
     "set_input\0"
     "share_params\0"
-    "get_input_name\0",
+    "get_input_name\0"
+    "get_output_index\0"
+    "get_output_info\0",
     aot_executor_registry_funcs};
 
 tvm_crt_error_t TVMAotExecutorModule_Register() {

diff --git a/src/runtime/crt/graph_executor_module/graph_executor_module.c b/src/runtime/crt/graph_executor_module/graph_executor_module.c
@@ -235,6 +235,8 @@ static const TVMBackendPackedCFunc graph_executor_registry_funcs[] = {
     &TVMGraphExecutorModule_Run,
     &TVMGraphExecutorModule_SetInput,
     &TVMGraphExecutorModule_NotImplemented,  // share_params
+    &TVMGraphExecutorModule_NotImplemented,  // get_output_index
+    &TVMGraphExecutorModule_NotImplemented,  // get_output_info
 };
 
 static const TVMFuncRegistry graph_executor_registry = {
@@ -247,7 +249,9 @@ static const TVMFuncRegistry graph_executor_registry = {
     "load_params\0"
     "run\0"
     "set_input\0"
-    "share_params\0",
+    "share_params\0"
+    "get_output_index\0"
+    "get_output_info\0",
     graph_executor_registry_funcs};
 
 tvm_crt_error_t TVMGraphExecutorModule_Register() {