Fix Windows MSVC arch detection (nomic-ai#1194)

- in llmodel.cpp to fix AVX-only handling Signed-off-by: cosmic-snow <[email protected]>
DEVBOX10 · Jul 13, 2023 · 6200900 · 6200900
1 parent 4963db8
commit 6200900
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/gpt4all-backend/llmodel.cpp b/gpt4all-backend/llmodel.cpp
@@ -18,7 +18,7 @@
 std::string s_implementations_search_path = ".";
 
 static bool has_at_least_minimal_hardware() {
-#ifdef __x86_64__
+#if defined(__x86_64__) || defined(_M_X64)
     #ifndef _MSC_VER
         return __builtin_cpu_supports("avx");
     #else
@@ -30,7 +30,7 @@ static bool has_at_least_minimal_hardware() {
 }
 
 static bool requires_avxonly() {
-#ifdef __x86_64__
+#if defined(__x86_64__) || defined(_M_X64)
     #ifndef _MSC_VER
         return !__builtin_cpu_supports("avx2");
     #else