Enabling exporting symbols on windows

Small fix to allow for the built library to be used in windows pytorch#728
philipNoonan · Jun 18, 2019 · 1dc8cde · 1dc8cde
1 parent c94a158
commit 1dc8cde
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/CMakeLists.txt b/CMakeLists.txt
@@ -8,6 +8,7 @@ file(GLOB_RECURSE HEADERS torchvision/csrc/vision.h)
 file(GLOB_RECURSE MODELS_HEADERS torchvision/csrc/models/*.h)
 file(GLOB_RECURSE MODELS_SOURCES torchvision/csrc/models/*.h torchvision/csrc/models/*.cpp)
 
+set(CMAKE_WINDOWS_EXPORT_ALL_SYMBOLS ON)
 add_library (${PROJECT_NAME} SHARED ${MODELS_SOURCES})
 target_link_libraries(${PROJECT_NAME} "${TORCH_LIBRARIES}")