Add resource_handle to both TVM_DLL_EXPORT_TYPED_FUNC and TVM_DLL_EXP…

…ORT_PACKED_FUNC macros in packed_func.h. This is a patch PR for #7388. (#7343) Co-authored-by: JC Li <jinli@nvidia.com>
apache · Jan 27, 2021 · eeec538 · eeec538
1 parent 1e0d356
commit eeec538
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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<decltype(f)>::FType;         \