ggml : fix arch check in bf16_to_fp32

2024-11-04 13:24:47 +01:00 · 2024-11-04 13:24:47 +01:00 · 1ba978549d
commit 1ba978549d
parent ea02c753eb
1 changed files with 2 additions and 0 deletions
--- a/ggml/src/ggml.c
+++ b/ggml/src/ggml.c
@ -392,6 +392,8 @@ void ggml_bf16_to_fp32_row(const ggml_bf16_t * x, float * y, int64_t n) {
                                    16)));
        }
    }
+#endif
+#if defined(__AVX2__)
    if (ggml_cpu_has_avx2()) {
        for (; i + 8 <= n; i += 8) {
            _mm256_storeu_ps(y + i,