Archives/gpt4all
2
0
Fork 0
mirror of https://github.com/nomic-ai/gpt4all synced 2024-11-06 09:20:33 +00:00
Code Issues Packages Projects Releases Wiki Activity

Revamp hardware tester to print to stdout the result in single word.

Browse Source
This commit is contained in:
Adam Treat 2023-04-21 07:36:05 -04:00
parent aecac945dd
commit 1beaf3bbbd
1 changed files with 16 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

28
test_hw.cpp
View File

@ -5,21 +5,25 @@ int main(int argc, char *argv[])
{
static bool avx = __builtin_cpu_supports("avx");
static bool avx2 = __builtin_cpu_supports("avx2");
static bool avx512f = __builtin_cpu_supports("avx512f");
static bool avx512vbmi = __builtin_cpu_supports("avx512vbmi");
static bool avx512vnni = __builtin_cpu_supports("avx512vnni");
static bool fma = __builtin_cpu_supports("fma");
static bool sse3 = __builtin_cpu_supports("sse3");
static std::string s;
s = "My hardware supports:\n";
s += "AVX = " + std::to_string(avx) + "\n";
s += "AVX2 = " + std::to_string(avx2) + "\n";
s += "AVX512 = " + std::to_string(avx512f) + "\n";
s += "AVX512_VBMI = " + std::to_string(avx512vbmi) + "\n";
s += "AVX512_VNNI = " + std::to_string(avx512vnni) + "\n";
s += "FMA = " + std::to_string(fma) + "\n";
s += "SSE3 = " + std::to_string(sse3) + "\n";
printf("%s", s.c_str());
s = "gpt4all hardware test results:\n";
s += " AVX = " + std::to_string(avx) + "\n";
s += " AVX2 = " + std::to_string(avx2) + "\n";
s += " FMA = " + std::to_string(fma) + "\n";
s += " SSE3 = " + std::to_string(sse3) + "\n";
fprintf(stderr, "%s", s.c_str());
fprintf(stderr, "your hardware supports the \"");
fflush(stderr);
if (avx2)
printf("full");
else if (avx && fma)
printf("avx_only");
else
printf("bare_minimum");
fflush(stdout);
fprintf(stderr, "\" version of gpt4all.\n");
fflush(stderr);
return 0;
}