[Openmp-commits] [openmp] [openmp] __kmp_x86_cpuid fix for i386/PIC builds. (PR #84626)

[David CARLIER via Openmp-commits]

https://github.com/devnexen created https://github.com/llvm/llvm-project/pull/84626

None

>From 17b22a36fad565829ca1490ada173fdb78708d32 Mon Sep 17 00:00:00 2001
From: David Carlier <devnexen at gmail.com>
Date: Sat, 9 Mar 2024 12:19:36 +0000
Subject: [PATCH] [openmp] __kmp_x86_cpuid fix for i386/PIC builds.

---
 openmp/runtime/src/kmp.h | 10 ++++++++++
 1 file changed, 10 insertions(+)

diff --git a/openmp/runtime/src/kmp.h b/openmp/runtime/src/kmp.h
index 1fc31779a217e5..ff10e6d69d09bb 100644
--- a/openmp/runtime/src/kmp.h
+++ b/openmp/runtime/src/kmp.h
@@ -1404,9 +1404,19 @@ extern void __kmp_query_cpuid(kmp_cpuinfo_t *p);
 // subleaf is only needed for cache and topology discovery and can be set to
 // zero in most cases
 static inline void __kmp_x86_cpuid(int leaf, int subleaf, struct kmp_cpuid *p) {
+#if defined(KMP_ARCH_X86_64) && (defined(__PIC__) || defined(__PIC__))
+// on i386 arch, the ebx reg. is used by pic, thus we need to preserve from
+// being trashed beforehand
+  __asm__ __volatile__("mov %%ebx, %%edi\n"
+		       "cpuid\n"
+		       "xchg %%edi, %%ebx\n"
+                       : "=a"(p->eax), "=b"(p->ebx), "=c"(p->ecx), "=d"(p->edx)
+                       : "a"(leaf), "c"(subleaf));
+#else
   __asm__ __volatile__("cpuid"
                        : "=a"(p->eax), "=b"(p->ebx), "=c"(p->ecx), "=d"(p->edx)
                        : "a"(leaf), "c"(subleaf));
+#endif
 }
 // Load p into FPU control word
 static inline void __kmp_load_x87_fpu_control_word(const kmp_int16 *p) {