pythongh-112535: Update _Py_ThreadId() to support RISC-V (pythongh-11…

…3084) Update _Py_ThreadId() to support RISC-V
brandtbucher · Dec 14, 2023 · f34e22c · f34e22c
1 parent 5f7d735
commit f34e22c
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/Include/object.h b/Include/object.h
@@ -283,6 +283,13 @@ _Py_ThreadId(void)
     // Both GCC and Clang have supported __builtin_thread_pointer
     // for s390 from long time ago.
     tid = (uintptr_t)__builtin_thread_pointer();
+#elif defined(__riscv)
+    #if defined(__clang__) && _Py__has_builtin(__builtin_thread_pointer)
+    tid = (uintptr_t)__builtin_thread_pointer();
+    #else
+    // tp is Thread Pointer provided by the RISC-V ABI.
+    __asm__ ("mv %0, tp" : "=r" (tid));
+    #endif
 #else
   # error "define _Py_ThreadId for this platform"
 #endif