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

Merge pull request #4465 from XiWeiGu/utest-zscal 

utest: Add tests for zscal
This commit is contained in:
Martin Kroeker 2024-01-31 14:19:19 +01:00 committed by GitHub
parent 98c9ff3194 969601a1dc
commit d1343302bd
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 100 additions and 43 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
kernel/x86_64/zscal.c
View File

@ -254,9 +254,15 @@ int CNAME(BLASLONG n, BLASLONG dummy0, BLASLONG dummy1, FLOAT da_r, FLOAT da_i,
while(j < n1) while(j < n1)
{ {
if (isnan(x[i]) || isinf(x[i]))
temp0 = NAN;
else
temp0 = -da_i * x[i+1]; temp0 = -da_i * x[i+1];
x[i+1] = da_i * x[i]; x[i+1] = da_i * x[i];
x[i] = temp0; x[i] = temp0;
if (isnan(x[i+inc_x]) || isinf(x[i+inc_x]))
temp1 = NAN;
else
temp1 = -da_i * x[i+1+inc_x]; temp1 = -da_i * x[i+1+inc_x];
x[i+1+inc_x] = da_i * x[i+inc_x]; x[i+1+inc_x] = da_i * x[i+inc_x];
x[i+inc_x] = temp1; x[i+inc_x] = temp1;
@ -268,6 +274,9 @@ int CNAME(BLASLONG n, BLASLONG dummy0, BLASLONG dummy1, FLOAT da_r, FLOAT da_i,
while(j < n) while(j < n)
{ {
if (isnan(x[i]) || isinf(x[i]))
temp0 = NAN;
else
temp0 = -da_i * x[i+1]; temp0 = -da_i * x[i+1];
x[i+1] = da_i * x[i]; x[i+1] = da_i * x[i];
x[i] = temp0; x[i] = temp0;

48
utest/test_zscal.c
View File

@ -20,6 +20,18 @@ CTEST(zscal, i_nan)
ASSERT_TRUE(isnan(nan[17])); ASSERT_TRUE(isnan(nan[17]));
} }
CTEST(zscal, i_nan_inc_2)
{
double i[] = {0,1, 0,1, 0,1, 0,1, 0,1, 0,1, 0,1, 0,1, 0,1 };
double nan[] = {NAN,0, NAN,0, NAN,0, NAN,0, NAN,0, NAN,0, NAN,0, NAN,0, NAN,0, NAN,0,
NAN,0, NAN,0, NAN,0, NAN,0, NAN,0, NAN,0, NAN,0, NAN,0, NAN,0, NAN,0};
cblas_zscal(9, i, &nan, 2);
ASSERT_TRUE(isnan(nan[0]));
ASSERT_TRUE(isnan(nan[1]));
ASSERT_TRUE(isnan(nan[16]));
ASSERT_TRUE(isnan(nan[17]));
}
CTEST(zscal, nan_i) CTEST(zscal, nan_i)
{ {
double i[] = {0,1, 0,1, 0,1, 0,1, 0,1, 0,1, 0,1, 0,1, 0,1 }; double i[] = {0,1, 0,1, 0,1, 0,1, 0,1, 0,1, 0,1, 0,1, 0,1 };
@ -31,6 +43,18 @@ CTEST(zscal, nan_i)
ASSERT_TRUE(isnan(i[17])); ASSERT_TRUE(isnan(i[17]));
} }
CTEST(zscal, nan_i_inc_2)
{
double i[] = {0,1, 0,1, 0,1, 0,1, 0,1, 0,1, 0,1, 0,1, 0,1,
0,1, 0,1, 0,1, 0,1, 0,1, 0,1, 0,1, 0,1, 0,1 };
double nan[] = {NAN,0, NAN,0, NAN,0, NAN,0, NAN,0, NAN,0, NAN,0, NAN,0, NAN,0, NAN,0};
cblas_zscal(9, &nan, &i, 2);
ASSERT_TRUE(isnan(i[0]));
ASSERT_TRUE(isnan(i[1]));
ASSERT_TRUE(isnan(i[16]));
ASSERT_TRUE(isnan(i[17]));
}
CTEST(zscal, i_inf) CTEST(zscal, i_inf)
{ {
double i[] = {0,1, 0,1, 0,1, 0,1, 0,1, 0,1, 0,1, 0,1, 0,1 }; double i[] = {0,1, 0,1, 0,1, 0,1, 0,1, 0,1, 0,1, 0,1, 0,1 };
@ -42,6 +66,18 @@ CTEST(zscal, i_inf)
ASSERT_TRUE(isinf(inf[17])); ASSERT_TRUE(isinf(inf[17]));
} }
CTEST(zscal, i_inf_inc_2)
{
double i[] = {0,1, 0,1, 0,1, 0,1, 0,1, 0,1, 0,1, 0,1, 0,1 };
double inf[] = {INFINITY, 0, INFINITY,0, INFINITY,0, INFINITY,0, INFINITY,0, INFINITY,0, INFINITY,0, INFINITY,0, INFINITY,0,
INFINITY, 0, INFINITY,0, INFINITY,0, INFINITY,0, INFINITY,0, INFINITY,0, INFINITY,0, INFINITY,0, INFINITY,0};
cblas_zscal(9, i, &inf, 2);
ASSERT_TRUE(isnan(inf[0]));
ASSERT_TRUE(isinf(inf[1]));
ASSERT_TRUE(isnan(inf[16]));
ASSERT_TRUE(isinf(inf[17]));
}
CTEST(zscal, inf_i) CTEST(zscal, inf_i)
{ {
double i[] = {0,1, 0,1, 0,1, 0,1, 0,1, 0,1, 0,1, 0,1, 0,1 }; double i[] = {0,1, 0,1, 0,1, 0,1, 0,1, 0,1, 0,1, 0,1, 0,1 };
@ -53,4 +89,16 @@ CTEST(zscal, inf_i)
ASSERT_TRUE(isinf(i[17])); ASSERT_TRUE(isinf(i[17]));
} }
CTEST(zscal, inf_i_inc_2)
{
double i[] = {0,1, 0,1, 0,1, 0,1, 0,1, 0,1, 0,1, 0,1, 0,1,
0,1, 0,1, 0,1, 0,1, 0,1, 0,1, 0,1, 0,1, 0,1 };
double inf[] = {INFINITY, 0, INFINITY,0, INFINITY,0, INFINITY,0, INFINITY,0, INFINITY,0, INFINITY,0, INFINITY,0, INFINITY,0};
cblas_zscal(9, &inf, &i, 2);
ASSERT_TRUE(isnan(i[0]));
ASSERT_TRUE(isinf(i[1]));
ASSERT_TRUE(isnan(i[16]));
ASSERT_TRUE(isinf(i[17]));
}
#endif #endif