From 51d8a8e1fc0115f17ec0b35b128e117bd54cce24 Mon Sep 17 00:00:00 2001 From: JC Li Date: Tue, 26 Jan 2021 14:38:24 -0800 Subject: [PATCH] Add resource_handle to both TVM_DLL_EXPORT_TYPED_FUNC and TVM_DLL_EXPORT_PACKED_FUNC macros in packed_func.h. This is a patch PR for #7388. --- include/tvm/runtime/packed_func.h | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/include/tvm/runtime/packed_func.h b/include/tvm/runtime/packed_func.h index ee2c7fd830f0..fd4e2114b11a 100644 --- a/include/tvm/runtime/packed_func.h +++ b/include/tvm/runtime/packed_func.h @@ -963,7 +963,7 @@ struct PackedFuncValueConverter { #define TVM_DLL_EXPORT_TYPED_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) { \ try { \ auto f = Function; \ using FType = ::tvm::runtime::detail::function_signature::FType; \