Fix Windows build for Neo DLR

wweic · Feb 5, 2019 · 7d96118 · 7d96118
1 parent cedebf7
commit 7d96118
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git 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
@@ -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);