gh-112535: Update _Py_ThreadId() to support RISC-V (gh-113084)

furkanonder · web-flow · commit f34e22c6470d · 2023-12-15T08:42:33.000+09:00
Update _Py_ThreadId() to support RISC-V
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
