Procházet zdrojové kódy

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

Przemysław Pawełczyk před 1 rokem
rodič
revize
182adefcf3
1 změnil soubory, kde provedl 7 přidání a 0 odebrání
  1. 7 0
      ggml-impl.h

+ 7 - 0
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