Export only on Apple platforms

cyyever · cyyever · commit 20fae6fc98ac · 2024-12-15T21:42:38.000+08:00
diff --git a/torch/CMakeLists.txt b/torch/CMakeLists.txt
@@ -310,6 +310,10 @@ endif()
 
 add_library(torch_python SHARED ${TORCH_PYTHON_SRCS})
 torch_compile_options(torch_python)  # see cmake/public/utils.cmake
+if(APPLE)
+  target_compile_options(torch_python PRIVATE
+      $<$<COMPILE_LANGUAGE:CXX>: -fvisibility=default>)
+endif()
 
 if(CAFFE2_USE_MKL AND BUILD_LIBTORCHLESS)
 
diff --git a/torch/csrc/Module.cpp b/torch/csrc/Module.cpp
@@ -46,6 +46,7 @@
 #include <torch/csrc/Dtype.h>
 #include <torch/csrc/DynamicTypes.h>
 #include <torch/csrc/Event.h>
+#include <torch/csrc/Export.h>
 #include <torch/csrc/Generator.h>
 #include <torch/csrc/Layout.h>
 #include <torch/csrc/MemoryFormat.h>
@@ -1717,7 +1718,7 @@ class WeakTensorRef {
   }
 };
 
-extern "C" C10_EXPORT PyObject* initModule();
+extern "C" TORCH_PYTHON_API PyObject* initModule();
 // separate decl and defn for msvc error C2491
 PyObject* initModule() {
   HANDLE_TH_ERRORS
