vrr/koboldcpp
2
0
Fork 0
mirror of https://github.com/LostRuins/koboldcpp.git synced 2025-09-14 10:59:41 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

better way of checking for avx2 support

Browse source
This commit is contained in:
Concedo 2025-06-22 22:56:50 +08:00
parent 52dcfe42d6
commit abc1d8ac25
14 changed files with 106 additions and 40 deletions
Download patch file Download diff file Expand all files Collapse all files

43
simplecpuinfo.cpp Normal file
View file

@ -0,0 +1,43 @@
//a simple program that obtains AVX and AVX2 info, prints them out and exits
#include <cstdio>
#if defined(_MSC_VER)
#include <intrin.h>
#elif defined(__GNUC__)
#include <cpuid.h>
#endif
bool check_avx_support() {
unsigned int cpuInfo[4] = {0, 0, 0, 0};
// Get CPU features
#if defined(_MSC_VER)
__cpuid(reinterpret_cast<int*>(cpuInfo), 1);
#elif defined(__GNUC__)
__cpuid(1, cpuInfo[0], cpuInfo[1], cpuInfo[2], cpuInfo[3]);
#endif
// Check AVX support (bit 28 of ECX)
return (cpuInfo[2] & (1 << 28)) != 0;
}
bool check_avx2_support() {
unsigned int cpuInfo[4] = {0, 0, 0, 0};
// Get extended CPU features
#if defined(_MSC_VER)
__cpuidex(reinterpret_cast<int*>(cpuInfo), 7, 0);
#elif defined(__GNUC__)
__cpuid_count(7, 0, cpuInfo[0], cpuInfo[1], cpuInfo[2], cpuInfo[3]);
#endif
// Check AVX2 support (bit 5 of EBX)
return (cpuInfo[1] & (1 << 5)) != 0;
}
int main() {
int avxSupported = check_avx_support()?1:0;
int avx2Supported = check_avx2_support()?1:0;
printf("{\"avx\":%d, \"avx2\":%d}",avxSupported,avx2Supported);
fflush(stdout);
return 0;
}