@@ -121,12 +121,12 @@ FORCEINLINE void copy_x(BLASLONG n, IFLOAT *src, IFLOAT *dest, BLASLONG inc_src)
121
121
122
122
FORCEINLINE void copy_y_beta (BLASLONG n , FLOAT * src , FLOAT * dest , BLASLONG inc_src , FLOAT beta )
123
123
{
124
- if (beta == 0 ) {
124
+ if (beta == ( FLOAT ) 0 ) {
125
125
for (BLASLONG i = 0 ; i < n ; i ++ ) {
126
126
* dest ++ = (FLOAT )0 ;
127
127
src += inc_src ;
128
128
}
129
- } else if (beta == 1 ) {
129
+ } else if (beta == ( FLOAT ) 1 ) {
130
130
for (BLASLONG i = 0 ; i < n ; i ++ ) {
131
131
* dest ++ = * src ;
132
132
src += inc_src ;
@@ -141,12 +141,12 @@ FORCEINLINE void copy_y_beta(BLASLONG n, FLOAT *src, FLOAT *dest, BLASLONG inc_s
141
141
142
142
FORCEINLINE void copy_y (BLASLONG n , FLOAT * src , FLOAT * dest , BLASLONG inc_src , FLOAT beta )
143
143
{
144
- if (beta == 0 ) {
144
+ if (beta == ( FLOAT ) 0 ) {
145
145
for (BLASLONG i = 0 ; i < n ; i ++ ) {
146
146
* dest = * src ++ ;
147
147
dest += inc_src ;
148
148
}
149
- } else if (beta == 1 ) {
149
+ } else if (beta == ( FLOAT ) 1 ) {
150
150
for (BLASLONG i = 0 ; i < n ; i ++ ) {
151
151
* dest += * src ++ ;
152
152
dest += inc_src ;
@@ -169,9 +169,9 @@ FORCEINLINE void move_y(BLASLONG n, FLOAT *src, FLOAT *dest, BLASLONG inc_dest)
169
169
170
170
static void BF16GEMV_N_beta (BLASLONG n , FLOAT * output_vector , FLOAT * input_vector , FLOAT beta )
171
171
{
172
- if (beta == 0 ) {
172
+ if (beta == ( FLOAT ) 0 ) {
173
173
memset (output_vector , 0 , sizeof (FLOAT ) * n );
174
- } else if (beta == 1 ) {
174
+ } else if (beta == ( FLOAT ) 1 ) {
175
175
if (output_vector != input_vector ) {
176
176
memcpy (output_vector , input_vector , sizeof (FLOAT ) * n );
177
177
}
0 commit comments