Skip to content

Commit f46e7ee

Browse files
chore: update bench.c to handle NaN elements
1 parent 2fa0ec8 commit f46e7ee

File tree

1 file changed

+10
-2
lines changed

1 file changed

+10
-2
lines changed

lib/node_modules/@stdlib/stats/base/dnanmeanpn/benchmark/c/benchmark.length.c

Lines changed: 10 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -102,7 +102,11 @@ static double benchmark1( int iterations, int len ) {
102102
int i;
103103

104104
for ( i = 0; i < len; i++ ) {
105-
x[ i ] = ( rand_double() * 20000.0 ) - 10000.0;
105+
if ( rand_double() < 0.2 ) {
106+
x[ i ] = 0.0 / 0.0; // NaN
107+
} else {
108+
x[ i ] = ( rand_double()*20000.0 ) - 10000.0;
109+
}
106110
}
107111
v = 0.0;
108112
t = tic();
@@ -136,7 +140,11 @@ static double benchmark2( int iterations, int len ) {
136140
int i;
137141

138142
for ( i = 0; i < len; i++ ) {
139-
x[ i ] = ( rand_double() * 20000.0 ) - 10000.0;
143+
if ( rand_double() < 0.2 ) {
144+
x[ i ] = 0.0 / 0.0; // NaN
145+
} else {
146+
x[ i ] = ( rand_double()*20000.0 ) - 10000.0;
147+
}
140148
}
141149
v = 0.0;
142150
t = tic();

0 commit comments

Comments
 (0)