hypergrep/src/cpu_features.cpp at master · p-ranav/hypergrep · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
#include <cpuid.h>
#include <hypergrep/cpu_features.hpp>

bool has_cpuid() {
  unsigned int eax, ebx, ecx, edx;
  __cpuid(0, eax, ebx, ecx, edx);
  return ebx == 0x756e6547 && edx == 0x49656e69 && ecx == 0x6c65746e;
}

bool has_cpuid_extension() {
  unsigned int eax, ebx, ecx, edx;
  __cpuid(1, eax, ebx, ecx, edx);
  return ecx & bit_SSE4_2;
}

bool has_avx2_support() {
  if (has_cpuid() && has_cpuid_extension()) {
    unsigned int eax, ebx, ecx, edx;
    if (__get_cpuid(1, &eax, &ebx, &ecx, &edx)) {
      if (ecx & bit_AVX2) {
        return true; // AVX2 is supported
      }
    }
  }
  return false; // AVX2 is not supported
}

// Function to check if the CPU supports AVX-512
bool has_avx512_support() {
  if (has_cpuid() && has_cpuid_extension()) {
    unsigned int eax, ebx, ecx, edx;
    if (__get_cpuid(0x00000007, &eax, &ebx, &ecx, &edx)) {
      if (ebx & bit_AVX512F && ebx & bit_AVX512DQ) {
        return true; // AVX-512 is supported
      }
    }
  }
  return false; // AVX-512 is not supported
}

// Function to check if the CPU supports AVX-512VBMI
bool has_avx512vbmi_support() {
  if (has_cpuid() && has_cpuid_extension()) {
    unsigned int eax, ebx, ecx, edx;
    if (__get_cpuid(0x00000007, &eax, &ebx, &ecx, &edx)) {
      if (ebx & bit_AVX512VBMI) {
        return true; // AVX-512VBMI is supported
      }
    }
  }
  return false; // AVX-512VBMI is not supported
}