Spaces:
Running
Running
Diego Devesa
commited on
Commit
·
09e4a9b
1
Parent(s):
b4c65b4
ggml : fix arch check in bf16_to_fp32 (llama/10164)
Browse files- 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,
|