Diego Devesa commited on
Commit
09e4a9b
·
1 Parent(s): b4c65b4

ggml : fix arch check in bf16_to_fp32 (llama/10164)

Browse files
Files changed (1) hide show
  1. ggml/src/ggml.c +2 -0
ggml/src/ggml.c CHANGED
@@ -395,6 +395,8 @@ void ggml_bf16_to_fp32_row(const ggml_bf16_t * x, float * y, int64_t n) {
395
  16)));
396
  }
397
  }
 
 
398
  if (ggml_cpu_has_avx2()) {
399
  for (; i + 8 <= n; i += 8) {
400
  _mm256_storeu_ps(y + i,
 
395
  16)));
396
  }
397
  }
398
+ #endif
399
+ #if defined(__AVX2__)
400
  if (ggml_cpu_has_avx2()) {
401
  for (; i + 8 <= n; i += 8) {
402
  _mm256_storeu_ps(y + i,