Skip to content

Commit 958f721

Browse files
committed
Beta fix for generic gemv T
1 parent 578e7da commit 958f721

File tree

1 file changed

+8
-1
lines changed

1 file changed

+8
-1
lines changed

kernel/generic/gemv_t.c

Lines changed: 8 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -52,7 +52,14 @@ int CNAME(BLASLONG m, BLASLONG n, FLOAT alpha, IFLOAT *a, BLASLONG lda, IFLOAT *
5252
temp += BF16TOF32(a_ptr[i]) * BF16TOF32(x[ix]);
5353
ix += inc_x;
5454
}
55-
y[iy] = F32TOBF16(BF16TOF32(y[iy]) + (ALPHA * temp));
55+
if (BETA == ZERO)
56+
{
57+
y[iy] = F32TOBF16(ALPHA * temp);
58+
}
59+
else
60+
{
61+
y[iy] = F32TOBF16(ALPHA * temp + BETA * BF16TOF32(y[iy]));
62+
}
5663
iy += inc_y;
5764
a_ptr += lda;
5865
}

0 commit comments

Comments
 (0)