diff --git a/include/tvm/runtime/packed_func.h b/include/tvm/runtime/packed_func.h index ee4ab82cd4d3..ee2c7fd830f0 100644 --- a/include/tvm/runtime/packed_func.h +++ b/include/tvm/runtime/packed_func.h @@ -910,9 +910,9 @@ struct PackedFuncValueConverter { #define TVM_DLL_EXPORT_PACKED_FUNC(ExportName, Function) \ extern "C" { \ TVM_DLL int ExportName(TVMValue* args, int* type_code, int num_args, TVMValue* out_value, \ - int* out_type_code); \ + int* out_type_code, void* resource_handle); \ int ExportName(TVMValue* args, int* type_code, int num_args, TVMValue* out_value, \ - int* out_type_code) { \ + int* out_type_code, void* resource_handle) { \ try { \ ::tvm::runtime::TVMRetValue rv; \ Function(::tvm::runtime::TVMArgs(args, type_code, num_args), &rv); \