floraachy
/
OpenBLAS
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Update zscal_vector.c

This commit is contained in:
Martin Kroeker 2024-01-07 19:45:28 +01:00 committed by GitHub
parent b08a208365
commit 2173356d5b
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 21 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

24
kernel/riscv64/zscal_vector.c
View File

@ -59,28 +59,8 @@ int CNAME(BLASLONG n, BLASLONG dummy0, BLASLONG dummy1, FLOAT da_r,FLOAT da_i, F
unsigned int gvl = 0; unsigned int gvl = 0;
FLOAT_V_T vt, v0, v1; FLOAT_V_T vt, v0, v1;
if (isnan(da_r)) {
gvl = VSETVL(n);
BLASLONG stride_x = inc_x * 2 * sizeof(FLOAT);
BLASLONG inc_xv = inc_x * 2 * gvl;
vt = VFMVVF_FLOAT(da_r, gvl);
for(i=0,j=0; i < n/(gvl*2); i++){
VSSEV_FLOAT(&x[ix], stride_x, vt, gvl);
VSSEV_FLOAT(&x[ix+1], stride_x, vt, gvl);
VSSEV_FLOAT(&x[ix+inc_xv], stride_x, vt, gvl);
VSSEV_FLOAT(&x[ix+inc_xv+1], stride_x, vt, gvl);
j += gvl*2; if(da_r == 0.0 && da_i == 0.0){
ix += inc_xv*2;
}
for(; j < n; ){
gvl = VSETVL(n-j);
VSSEV_FLOAT(&x[ix], stride_x, vt, gvl);
VSSEV_FLOAT(&x[ix+1], stride_x, vt, gvl);
j += gvl;
ix += inc_x * 2 * gvl;
}
} else if(da_r == 0.0 && da_i == 0.0){
gvl = VSETVL(n); gvl = VSETVL(n);
BLASLONG stride_x = inc_x * 2 * sizeof(FLOAT); BLASLONG stride_x = inc_x * 2 * sizeof(FLOAT);
BLASLONG inc_xv = inc_x * 2 * gvl; BLASLONG inc_xv = inc_x * 2 * gvl;
@ -101,6 +81,7 @@ int CNAME(BLASLONG n, BLASLONG dummy0, BLASLONG dummy1, FLOAT da_r,FLOAT da_i, F
j += gvl; j += gvl;
ix += inc_x * 2 * gvl; ix += inc_x * 2 * gvl;
} }
#if 0
}else if(da_r == 0.0){ }else if(da_r == 0.0){
gvl = VSETVL(n); gvl = VSETVL(n);
BLASLONG stride_x = inc_x * 2 * sizeof(FLOAT); BLASLONG stride_x = inc_x * 2 * sizeof(FLOAT);
@ -129,6 +110,7 @@ int CNAME(BLASLONG n, BLASLONG dummy0, BLASLONG dummy1, FLOAT da_r,FLOAT da_i, F
VSSEV_FLOAT(&x[ix], stride_x, vt, gvl); VSSEV_FLOAT(&x[ix], stride_x, vt, gvl);
VSSEV_FLOAT(&x[ix+1], stride_x, v1, gvl); VSSEV_FLOAT(&x[ix+1], stride_x, v1, gvl);
} }
#endif
}else if(da_i == 0.0){ }else if(da_i == 0.0){
gvl = VSETVL(n); gvl = VSETVL(n);
BLASLONG stride_x = inc_x * 2 * sizeof(FLOAT); BLASLONG stride_x = inc_x * 2 * sizeof(FLOAT);