Sfoglia il codice sorgente

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

Przemysław Pawełczyk 1 anno fa
parent
commit
182adefcf3
1 ha cambiato i file con 7 aggiunte e 0 eliminazioni
  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