Disable thread-local tape for python builds

auto-differentiation · Apr 4, 2024 · a33a5b1 · a33a5b1
1 parent f3c3dc1
commit a33a5b1
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 10 deletions.
diff --git a/build_extensions.py b/build_extensions.py
@@ -41,15 +41,6 @@
     from distutils.extension import Extension as _Extension  # type: ignore[assignment]
 
 
-# Convert distutils Windows platform specifiers to CMake -A arguments
-PLAT_TO_CMAKE = {
-    "win32": "Win32",
-    "win-amd64": "x64",
-    "win-arm32": "ARM",
-    "win-arm64": "ARM64",
-}
-
-
 def get_vsvars_environment(architecture="amd64", toolset="14.3"):
     """Returns a dictionary containing the environment variables set up by vsvarsall.bat
     architecture - Architecture to pass to vcvarsall.bat. Normally "x86" or "amd64"

diff --git a/pyproject.toml b/pyproject.toml
@@ -24,7 +24,7 @@
 
 [tool.poetry]
 name = "xad"
-version = "1.5.1.2"
+version = "1.5.1.3"
 description = "High-Performance Automatic Differentiation for Python"
 authors = ["Auto Differentiation Dev Team <[email protected]>"]
 readme = "README.md"

diff --git a/src/CMakeLists.txt b/src/CMakeLists.txt
@@ -25,6 +25,7 @@
 if(MSVC)
   option(XAD_STATIC_MSVC_RUNTIME "Use static runtime" ON)
 endif()
+option(XAD_NO_THREADLOCAL "Use thread-local tape" ON)
 
 find_package(Threads REQUIRED)
 
@@ -48,6 +49,7 @@ pybind11_add_module(
 target_include_directories(_xad PRIVATE ${CMAKE_CURRENT_BINARY_DIR}/xad/src
                                         xad/src)
 set_target_properties(_xad PROPERTIES CXX_STANDARD 17)
+target_link_libraries(_xad PRIVATE Threads::Threads)
 if(MSVC)
   target_compile_options(_xad PRIVATE -nologo -utf-8 "/arch:AVX")
   target_compile_definitions(