diff --git a/CMakeLists.txt b/CMakeLists.txt index 1b01bac8a657..fbbe2feeb9fc 100644 --- a/CMakeLists.txt +++ b/CMakeLists.txt @@ -326,5 +326,6 @@ if(MSVC) target_compile_definitions(tvm_runtime PRIVATE -DHalide_EXPORTS) target_compile_definitions(tvm PRIVATE -DTVM_EXPORTS) target_compile_definitions(tvm_runtime PRIVATE -DTVM_EXPORTS) + target_compile_definitions(tvm_runtime_static PRIVATE -DTVM_DLL= -DTVM_DLL_PARALLEL_LAUNCH=__declspec(dllexport)) target_compile_definitions(nnvm_compiler PRIVATE -DNNVM_EXPORTS) endif() diff --git a/include/tvm/runtime/c_backend_api.h b/include/tvm/runtime/c_backend_api.h index f55748e38289..b69697427ed4 100644 --- a/include/tvm/runtime/c_backend_api.h +++ b/include/tvm/runtime/c_backend_api.h @@ -104,7 +104,9 @@ typedef int (*FTVMParallelLambda)( * * \return 0 when no error is thrown, -1 when failure happens */ -TVM_DLL int TVMBackendParallelLaunch(FTVMParallelLambda flambda, +#define TVM_DLL_PARALLEL_LAUNCH + +TVM_DLL_PARALLEL_LAUNCH int TVMBackendParallelLaunch(FTVMParallelLambda flambda, void* cdata, int num_task);