python · corona10 · Nov 30, 2023 · vstinner · Nov 30, 2023 · vstinner
diff --git a/Include/object.h b/Include/object.h
@@ -261,6 +261,8 @@ _Py_ThreadId(void)
     __asm__ ("mrs %0, tpidrro_el0" : "=r" (tid));
 #elif defined(__aarch64__)
     __asm__ ("mrs %0, tpidr_el0" : "=r" (tid));
+#elif define(HAVE_BUILTIN_THREAD_POINTER)
+    tid = (uintptr_t)__builtin_thread_pointer();
 #else
   # error "define _Py_ThreadId for this platform"
 #endif

@@ -4541,6 +4541,23 @@ AS_VAR_IF([ac_cv_builtin_atomic], [yes], [
     AC_DEFINE(HAVE_BUILTIN_ATOMIC, 1, [Has builtin __atomic_load_n() and __atomic_store_n() functions])
 ])
 
+AC_CACHE_CHECK([for builtin __builtin_thread_pointer], [ac_cv_builtin_thread_pointer], [
+AC_LINK_IFELSE(
+[
+  AC_LANG_SOURCE([[
+    void *tp;
+    int main() {
+      tp = __builtin_thread_pointer();
+      return 0;
+    }
+  ]])
+],[ac_cv_builtin_thread_pointer=yes],[ac_cv_builtin_thread_pointer=no])
+])
+
+AS_VAR_IF([ac_cv_builtin_thread_pointer], [yes], [
+    AC_DEFINE(HAVE_BUILTIN_THREAD_POINTER, 1, [Has builtin __builtin_thread_pointer])
+])
+
 # --with-mimalloc
 AC_MSG_CHECKING([for --with-mimalloc])
 AC_ARG_WITH([mimalloc],

diff --git a/pyconfig.h.in b/pyconfig.h.in
@@ -127,6 +127,9 @@
 /* Has builtin __atomic_load_n() and __atomic_store_n() functions */
 #undef HAVE_BUILTIN_ATOMIC
 
+/* Has builtin __builtin_thread_pointer */
+#undef HAVE_BUILTIN_THREAD_POINTER
+
 /* Define to 1 if you have the <bzlib.h> header file. */
 #undef HAVE_BZLIB_H