Skip to content

Commit

Permalink
ggml : fix arch check in bf16_to_fp32 (ggerganov#10164)
Browse files Browse the repository at this point in the history
  • Loading branch information
slaren authored Nov 4, 2024
1 parent 3407364 commit a9e8a9a
Showing 1 changed file with 2 additions and 0 deletions.
2 changes: 2 additions & 0 deletions ggml/src/ggml.c
Original file line number Diff line number Diff line change
Expand Up @@ -395,6 +395,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,
Expand Down

0 comments on commit a9e8a9a

Please sign in to comment.