summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--ggml-quants.c2
1 files changed, 1 insertions, 1 deletions
diff --git a/ggml-quants.c b/ggml-quants.c
index 2eaca059..f26798ac 100644
--- a/ggml-quants.c
+++ b/ggml-quants.c
@@ -132,7 +132,7 @@ static inline __m256 sum_i16_pairs_float(const __m256i x) {
}
static inline __m256 mul_sum_us8_pairs_float(const __m256i ax, const __m256i sy) {
-#if __AVXVNNI__
+#if defined(__AVXVNNI__) || defined(__AVX512VNNI__)
const __m256i zero = _mm256_setzero_si256();
const __m256i summed_pairs = _mm256_dpbusd_epi32(zero, ax, sy);
return _mm256_cvtepi32_ps(summed_pairs);