AI/gpt4all
1
0
Fork 0
mirror of https://github.com/nomic-ai/gpt4all.git synced 2024-10-01 01:06:10 -04:00
Code Issues Packages Projects Releases Wiki Activity

windows/msvc: use high level processor feature detection API

Browse Source 
see https://learn.microsoft.com/en-us/windows/win32/api/processthreadsapi/nf-processthreadsapi-isprocessorfeaturepresent
This commit is contained in:
Aaron Miller 2023-07-12 11:44:08 -04:00 committed by AT
parent ad0e7fd01f
commit 95b8fb312e
1 changed files with 5 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
gpt4all-backend/llmodel.cpp
View File

@ -10,6 +10,9 @@
#include <cassert>
#include <cstdlib>
#include <sstream>
#ifdef _MSC_VER
#include <processthreadsapi.h>
#endif
std::string s_implementations_search_path = ".";
@ -18,9 +21,7 @@ static bool has_at_least_minimal_hardware() {
#ifndef _MSC_VER
return __builtin_cpu_supports("avx");
#else
int cpuInfo[4];
__cpuid(cpuInfo, 1);
return cpuInfo[2] & (1 << 28);
return IsProcessorFeaturePresent(PF_AVX_INSTRUCTIONS_AVAILABLE);
#endif
#else
return true; // Don't know how to handle non-x86_64
@ -32,9 +33,7 @@ static bool requires_avxonly() {
#ifndef _MSC_VER
return !__builtin_cpu_supports("avx2");
#else
int cpuInfo[4];
__cpuidex(cpuInfo, 7, 0);
return !(cpuInfo[1] & (1 << 5));
return !IsProcessorFeaturePresent(PF_AVX2_INSTRUCTIONS_AVAILABLE);
#endif
#else
return false; // Don't know how to handle non-x86_64