Dependencies

apache · Nov 12, 2024 · 4a81172 · 4a81172
1 parent 74ff9bb
commit 4a81172
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 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 = {
@@ -224,6 +226,8 @@ static const TVMFuncRegistry aot_executor_registry = {
     "set_input\0"
     "share_params\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 = {
@@ -248,6 +250,8 @@ static const TVMFuncRegistry graph_executor_registry = {
     "run\0"
     "set_input\0"
     "share_params\0",
+    "get_output_index\0",
+    "get_output_info\0",
     graph_executor_registry_funcs};
 
 tvm_crt_error_t TVMGraphExecutorModule_Register() {