Przeglądaj źródła

ggml : fix arch check in bf16_to_fp32 (#10164)

Diego Devesa 1 rok temu
rodzic
commit
a9e8a9a030
1 zmienionych plików z 2 dodań i 0 usunięć
  1. 2 0
      ggml/src/ggml.c

+ 2 - 0
ggml/src/ggml.c

@@ -395,6 +395,8 @@ void ggml_bf16_to_fp32_row(const ggml_bf16_t * x, float * y, int64_t n) {
                                     16)));
                                     16)));
         }
         }
     }
     }
+#endif
+#if defined(__AVX2__)
     if (ggml_cpu_has_avx2()) {
     if (ggml_cpu_has_avx2()) {
         for (; i + 8 <= n; i += 8) {
         for (; i + 8 <= n; i += 8) {
             _mm256_storeu_ps(y + i,
             _mm256_storeu_ps(y + i,