Wrap torch_python with torch_compile_options (#136743)

Fixes #ISSUE_NUMBER Pull Request resolved: #136743 Approved by: https://github.com/ezyang
pytorch · Nov 7, 2024 · 9d09af9 · 9d09af9
1 parent d0da40a
commit 9d09af9
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/torch/CMakeLists.txt b/torch/CMakeLists.txt
@@ -309,6 +309,11 @@ endif()
 
 
 add_library(torch_python SHARED ${TORCH_PYTHON_SRCS})
+torch_compile_options(torch_python)  # see cmake/public/utils.cmake
+if(NOT WIN32)
+  target_compile_options(torch_python PRIVATE
+      $<$<COMPILE_LANGUAGE:CXX>: -fvisibility=default>)
+endif()
 
 if(CAFFE2_USE_MKL AND BUILD_LIBTORCHLESS)