Skip to content

Commit a318c58

Browse files
committed
change quant.c
1 parent 1d47d63 commit a318c58

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

ggml/src/ggml-cpu/arch/arm/quants.c

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -45,7 +45,7 @@ void quantize_row_q8_0(const float * GGML_RESTRICT x, void * GGML_RESTRICT vy, i
4545
block_q8_0 * GGML_RESTRICT y = vy;
4646

4747
#if defined(__ARM_FEATURE_SVE)
48-
const int sve_register_length = ggml_cpu_get_sve_cnt() * 8;
48+
const int sve_register_length = svcntb() * 8;
4949
const int ggml_f32_epr = sve_register_length / 32;
5050
const svfloat32_t inactive1 = svdup_n_f32(0.0f);
5151
const svbool_t pg = svptrue_b32();
@@ -79,6 +79,7 @@ void quantize_row_q8_0(const float * GGML_RESTRICT x, void * GGML_RESTRICT vy, i
7979
svst1b_s32(pg, &y[i].qs[j], result);
8080
}
8181
}
82+
8283
#elif defined(__ARM_NEON)
8384
for (int i = 0; i < nb; i++) {
8485
float32x4_t srcv [8];

0 commit comments

Comments
 (0)