feature: add detection of cpu waitpkg support

Related-To: NEO-9737

Signed-off-by: Zbigniew Zdanowicz <[email protected]>

Author: zzdanowicz

shared/source/utilities/cpu_info.h

@@ -18,6 +18,7 @@
 namespace NEO {
 struct CpuInfo {
     static const uint64_t featureNone = 0x000000000ULL;
+    static const uint64_t featureWaitPkg = 0x000000001ULL;
     static const uint64_t featureAvX2 = 0x000800000ULL;
     static const uint64_t featureNeon = 0x001000000ULL;
     static const uint64_t featureClflush = 0x2000000000ULL;

shared/source/utilities/x86_64/cpu_info_x86_64.cpp

@@ -1,5 +1,5 @@
 /*
- * Copyright (C) 2021-2022 Intel Corporation
+ * Copyright (C) 2021-2023 Intel Corporation
  *
  * SPDX-License-Identifier: MIT
  *
@@ -13,31 +13,42 @@
 
 namespace NEO {
 void CpuInfo::detect() const {
+    constexpr uint32_t processorInfo = 1;
+    constexpr uint32_t extendedFeatures = 0x7;
+    constexpr uint32_t extendedFunctionId = 0x80000000;
+    constexpr uint32_t addressSize = 0x80000008;
+    constexpr size_t eax = 0;
+    constexpr size_t ebx = 1;
+    constexpr size_t ecx = 2;
+    constexpr size_t edx = 3;
+
     uint32_t cpuInfo[4] = {};
 
     cpuid(cpuInfo, 0u);
-    auto numFunctionIds = cpuInfo[0];
-    if (numFunctionIds >= 1u) {
-        cpuid(cpuInfo, 1u);
+    auto numFunctionIds = cpuInfo[eax];
+    if (numFunctionIds >= processorInfo) {
+        cpuid(cpuInfo, processorInfo);
         {
-            features |= cpuInfo[3] & BIT(19) ? featureClflush : featureNone;
+            features |= cpuInfo[edx] & BIT(19) ? featureClflush : featureNone;
         }
     }
 
-    if (numFunctionIds >= 7u) {
-        cpuid(cpuInfo, 7u);
+    if (numFunctionIds >= extendedFeatures) {
+        cpuid(cpuInfo, extendedFeatures);
         {
             auto mask = BIT(5) | BIT(3) | BIT(8);
-            features |= (cpuInfo[1] & mask) == mask ? featureAvX2 : featureNone;
+            features |= (cpuInfo[ebx] & mask) == mask ? featureAvX2 : featureNone;
+
+            features |= (cpuInfo[ecx] & BIT(5)) ? featureWaitPkg : featureNone;
         }
     }
 
-    cpuid(cpuInfo, 0x80000000);
-    auto maxExtendedId = cpuInfo[0];
-    if (maxExtendedId >= 0x80000008) {
-        cpuid(cpuInfo, 0x80000008);
+    cpuid(cpuInfo, extendedFunctionId);
+    auto maxExtendedId = cpuInfo[eax];
+    if (maxExtendedId >= addressSize) {
+        cpuid(cpuInfo, addressSize);
         {
-            virtualAddressSize = (cpuInfo[0] >> 8) & 0xFF;
+            virtualAddressSize = (cpuInfo[eax] >> 8) & 0xFF;
         }
     }
 }

shared/test/unit_test/utilities/x86_64/cpuinfo_tests_x86_64.cpp

@@ -51,6 +51,7 @@ TEST(CpuInfoTest, giveFunctionIsNotAvailableWhenFeatureIsNotSupportedThenMaskBit
 
     EXPECT_FALSE(testCpuInfo.isFeatureSupported(CpuInfo::featureAvX2));
     EXPECT_FALSE(testCpuInfo.isFeatureSupported(CpuInfo::featureClflush));
+    EXPECT_FALSE(testCpuInfo.isFeatureSupported(CpuInfo::featureWaitPkg));
 
     CpuInfo::cpuidFunc = defaultCpuidFunc;
 }
@@ -63,6 +64,7 @@ TEST(CpuInfoTest, giveFunctionIsAvailableWhenFeatureIsNotSupportedThenMaskBitIsO
 
     EXPECT_FALSE(testCpuInfo.isFeatureSupported(CpuInfo::featureAvX2));
     EXPECT_FALSE(testCpuInfo.isFeatureSupported(CpuInfo::featureClflush));
+    EXPECT_FALSE(testCpuInfo.isFeatureSupported(CpuInfo::featureWaitPkg));
 
     CpuInfo::cpuidFunc = defaultCpuidFunc;
 }
@@ -75,6 +77,7 @@ TEST(CpuInfoTest, whenFeatureIsSupportedThenMaskBitIsOn) {
 
     EXPECT_TRUE(testCpuInfo.isFeatureSupported(CpuInfo::featureAvX2));
     EXPECT_TRUE(testCpuInfo.isFeatureSupported(CpuInfo::featureClflush));
+    EXPECT_TRUE(testCpuInfo.isFeatureSupported(CpuInfo::featureWaitPkg));
 
     CpuInfo::cpuidFunc = defaultCpuidFunc;
 }