Skip to content

Commit 99f6158

Browse files
authored
Merge pull request #6127 from martin-frbg/issue6101
Use explicit zeroing on beta=0 in the non-unit stride path of ARM64 sbgemv_n too
2 parents 2059009 + 94b115e commit 99f6158

1 file changed

Lines changed: 8 additions & 1 deletion

File tree

‎kernel/arm64/sbgemv_n_neon.c‎

Lines changed: 8 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -494,10 +494,17 @@ int CNAME(BLASLONG m, BLASLONG n, FLOAT alpha, bfloat16 *a, BLASLONG lda,
494494
}
495495

496496
BLASLONG iy = 0;
497-
for (i = 0; i < m; i++) {
497+
if (beta == 0) {
498+
for (i = 0; i < m; i++) {
499+
y[iy] = 0.;
500+
iy += incy;
501+
}
502+
} else {
503+
for (i = 0; i < m; i++) {
498504
y[iy] *= beta;
499505
iy += incy;
500506
}
507+
}
501508

502509
for (j = 0; j < n; j++) {
503510
x0 = alpha * vcvtah_f32_bf16(*x_ptr);

0 commit comments

Comments
 (0)