[PATCH] D85531: [SystemZ/ZOS] Add support for getHostNumPhysicalCores()

[Kai Nacke via Phabricator via llvm-commits]

Kai updated this revision to Diff 284318.
Kai added a comment.

- Sorted condition according to OS
- Updated/fixed comment


CHANGES SINCE LAST ACTION
  https://reviews.llvm.org/D85531/new/

https://reviews.llvm.org/D85531

Files:
  llvm/lib/Support/Host.cpp
  llvm/unittests/Support/Host.cpp


Index: llvm/unittests/Support/Host.cpp
===================================================================
--- llvm/unittests/Support/Host.cpp
+++ llvm/unittests/Support/Host.cpp
@@ -37,12 +37,11 @@
   bool isSupportedArchAndOS() {
     // Initially this is only testing detection of the number of
     // physical cores, which is currently only supported/tested for
-    // x86_64 Linux and Darwin.
+    // Linux (x86_64, PPC64, SystemZ), Darwin (x86_64), and for z/OS.
     return (Host.isOSWindows() && llvm_is_multithreaded()) ||
-           (Host.isX86() &&
-            (Host.isOSDarwin() || Host.getOS() == Triple::Linux)) ||
-           (Host.getOS() == Triple::Linux &&
-            (Host.isPPC64() || Host.isSystemZ()));
+           (Host.isX86() && (Host.isOSDarwin() || Host.isOSLinux())) ||
+           (Host.isPPC64() && Host.isOSLinux()) ||
+           (Host.isSystemZ() && (Host.isOSLinux() || Host.isOSzOS()));
   }
 
   HostTest() : Host(Triple::normalize(sys::getProcessTriple())) {}
Index: llvm/lib/Support/Host.cpp
===================================================================
--- llvm/lib/Support/Host.cpp
+++ llvm/lib/Support/Host.cpp
@@ -1311,6 +1311,28 @@
   }
   return count;
 }
+#elif defined(__MVS__)
+int computeHostNumPhysicalCores() {
+  enum {
+    // Byte offset of the pointer to the Communications Vector Table (CVT) in
+    // the Prefixed Save Area (PSA). The table entry is a 31-bit pointer and
+    // will be zero-extended to uintptr_t.
+    FLCCVT = 16,
+    // Byte offset of the pointer to the Common System Data Area (CSD) in the
+    // CVT. The table entry is a 31-bit pointer and will be zero-extended to
+    // uintptr_t.
+    CVTCSD = 660,
+    // Byte offset to the number of live CPs in the LPAR, stored as a signed
+    // 32-bit value in the table.
+    CSD_NUMBER_ONLINE_STANDARD_CPS = 264,
+  };
+  char *PSA = 0;
+  char *CVT = reinterpret_cast<char *>(
+      static_cast<uintptr_t>(reinterpret_cast<unsigned int &>(PSA[FLCCVT])));
+  char *CSD = reinterpret_cast<char *>(
+      static_cast<uintptr_t>(reinterpret_cast<unsigned int &>(CVT[CVTCSD])));
+  return reinterpret_cast<int &>(CSD[CSD_NUMBER_ONLINE_STANDARD_CPS]);
+}
 #elif defined(_WIN32) && LLVM_ENABLE_THREADS != 0
 // Defined in llvm/lib/Support/Windows/Threading.inc
 int computeHostNumPhysicalCores();


-------------- next part --------------
A non-text attachment was scrubbed...
Name: D85531.284318.patch
Type: text/x-patch
Size: 2333 bytes
Desc: not available
URL: <http://lists.llvm.org/pipermail/llvm-commits/attachments/20200810/863d57ed/attachment.bin>