Skip to content

Commit c2b934d

Browse files
author
guozhong.zhuang
committed
New nancheck feature to support env variable LAPACKE_NANCHECK
1 parent 18518c8 commit c2b934d

1 file changed

Lines changed: 18 additions & 2 deletions

File tree

LAPACKE/src/lapacke_nancheck.c

Lines changed: 18 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -34,14 +34,30 @@
3434

3535
#include "lapacke_utils.h"
3636

37-
static lapack_int nancheck_flag = 1;
37+
static lapack_int nancheck_flag = -1;
38+
39+
40+
/* App can set this flag with API call */
41+
/* This flag is based on environment variable LAPACK_NANCHECK */
3842

3943
void LAPACKE_set_nancheck( lapack_int flag )
4044
{
41-
nancheck_flag = flag;
45+
nancheck_flag = ( flag ) ? 1 : 0;
4246
}
4347

4448
lapack_int LAPACKE_get_nancheck( )
4549
{
50+
if ( nancheck_flag != -1 )
51+
return nancheck_flag;
52+
53+
/* Check environment variable, once and only once */
54+
char* env = getenv("LAPACKE_NANCHECK");
55+
if ( !env ) {
56+
nancheck_flag = 1;
57+
} else {
58+
int envval = (int)atoi(env);
59+
nancheck_flag = ( envval ) ? 1 : 0;
60+
}
61+
4662
return nancheck_flag;
4763
}

0 commit comments

Comments
 (0)