Added free_threading option to adapt python headers and library to free threading (no GIL) mode

vfdev-5 · vfdev-5 · commit fee7a7246b6d · 2024-08-19T22:39:51.000+02:00
diff --git a/python/private/python_repositories.bzl b/python/private/python_repositories.bzl
@@ -118,6 +118,7 @@ def _python_repository_impl(rctx):
     python_version = rctx.attr.python_version
     python_version_info = python_version.split(".")
     python_short_version = "{0}.{1}".format(*python_version_info)
+    free_threading_postfix = "t" if rctx.attr.free_threading else ""
     release_filename = rctx.attr.release_filename
     urls = rctx.attr.urls or [rctx.attr.url]
     auth = get_auth(rctx, urls)
@@ -369,7 +370,7 @@ cc_library(
     hdrs = [":includes"],
     includes = [
         "include",
-        "include/python{python_version}",
+        "include/python{python_version}{ft_postfix}",
         "include/python{python_version}m",
     ],
 )
@@ -378,9 +379,9 @@ cc_library(
     name = "libpython",
     hdrs = [":includes"],
     srcs = select({{
-        "@platforms//os:windows": ["python3.dll", "libs/python{python_version_nodot}.lib"],
-        "@platforms//os:macos": ["lib/libpython{python_version}.dylib"],
-        "@platforms//os:linux": ["lib/libpython{python_version}.so", "lib/libpython{python_version}.so.1.0"],
+        "@platforms//os:windows": ["python3.dll", "libs/python{python_version_nodot}{ft_postfix}.lib"],
+        "@platforms//os:macos": ["lib/libpython{python_version}{ft_postfix}.dylib"],
+        "@platforms//os:linux": ["lib/libpython{python_version}{ft_postfix}.so", "lib/libpython{python_version}{ft_postfix}.so.1.0"],
     }}),
 )
 
@@ -432,6 +433,7 @@ py_exec_tools_toolchain(
         python_path = python_bin,
         python_version = python_short_version,
         python_version_nodot = python_short_version.replace(".", ""),
+        ft_postfix = free_threading_postfix,
         coverage_attr = coverage_attr_text,
         interpreter_version_info_major = python_version_info[0],
         interpreter_version_info_minor = python_version_info[1],
@@ -507,6 +509,11 @@ For more information see the official bazel docs
                   "Either distutils or distutils_content can be specified, but not both.",
             mandatory = False,
         ),
+        "free_threading": attr.bool(
+            default = False,
+            doc = "Whether python interpreter has enabled free-threading (no GIL) mode.",
+            mandatory = False,
+        ),
         "ignore_root_user_error": attr.bool(
             default = False,
             doc = "Whether the check for root should be ignored or not. This causes cache misses with .pyc files.",
