Merge pull request #1 from neo-ai/fix_win_build

Fix Windows build for Neo DLR
wweic · Feb 6, 2019 · 4477957 · 4477957
2 parents cedebf7 + 7d96118
commit 4477957
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);