Skip to content

Commit e56ba44

Browse files
slarenarthw
authored andcommitted
ggml : fix arch check in bf16_to_fp32 (ggml-org#10164)
1 parent 5af3304 commit e56ba44

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

ggml/src/ggml.c

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -395,6 +395,8 @@ void ggml_bf16_to_fp32_row(const ggml_bf16_t * x, float * y, int64_t n) {
395395
16)));
396396
}
397397
}
398+
#endif
399+
#if defined(__AVX2__)
398400
if (ggml_cpu_has_avx2()) {
399401
for (; i + 8 <= n; i += 8) {
400402
_mm256_storeu_ps(y + i,

0 commit comments

Comments
 (0)