summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--ggml-impl.h7
1 files changed, 7 insertions, 0 deletions
diff --git a/ggml-impl.h b/ggml-impl.h
index d85b152b..59684fa8 100644
--- 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