]> git.djapps.eu Git - pkg/ggml/sources/ggml/commitdiff
ggml : expose SSE3 and SSSE3 for MSVC when AVX is available (whisper/2128)
authorPrzemysław Pawełczyk <redacted>
Wed, 8 May 2024 15:33:43 +0000 (17:33 +0200)
committerGeorgi Gerganov <redacted>
Tue, 14 May 2024 12:31:03 +0000 (15:31 +0300)
src/ggml-impl.h

index d85b152bf16ea36988f98c26d2cb3a21c40a5e91..59684fa81f07e8d81010147bbf71b3842111fddd 100644 (file)
@@ -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