ggml : expose SSE3 and SSSE3 for MSVC when AVX is available (whisper/…

…2128)
heiner · May 14, 2024 · 182adef · 182adef
1 parent 0d26d8c
commit 182adef
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/ggml-impl.h b/ggml-impl.h
@@ -120,9 +120,16 @@ extern "C" {
 #ifndef __F16C__
 #define __F16C__
 #endif
+#endif
+
+// __SSE3__ and __SSSE3__ are not defined in MSVC, but SSE3/SSSE3 are present when AVX/AVX2/AVX512 are available
+#if defined(_MSC_VER) && (defined(__AVX__) || defined(__AVX2__) || defined(__AVX512F__))
 #ifndef __SSE3__
 #define __SSE3__
 #endif
+#ifndef __SSSE3__
+#define __SSSE3__
+#endif
 #endif
 
 // 16-bit float