LAPACK 3.11.0
LAPACK: Linear Algebra PACKage
Loading...
Searching...
No Matches

◆ LAPACKE_stf_nancheck()

lapack_logical LAPACKE_stf_nancheck ( int  matrix_layout,
char  transr,
char  uplo,
char  diag,
lapack_int  n,
const float *  a 
)

Definition at line 36 of file lapacke_stf_nancheck.c.

40{
41 lapack_int len;
42 lapack_logical rowmaj, ntr, lower, unit;
43 lapack_int n1, n2, k;
44
45 if( a == NULL ) return (lapack_logical) 0;
46
47 rowmaj = (matrix_layout == LAPACK_ROW_MAJOR);
48 ntr = LAPACKE_lsame( transr, 'n' );
49 lower = LAPACKE_lsame( uplo, 'l' );
50 unit = LAPACKE_lsame( diag, 'u' );
51
52 if( ( !rowmaj && ( matrix_layout != LAPACK_COL_MAJOR ) ) ||
53 ( !ntr && !LAPACKE_lsame( transr, 't' )
54 && !LAPACKE_lsame( transr, 'c' ) ) ||
55 ( !lower && !LAPACKE_lsame( uplo, 'u' ) ) ||
56 ( !unit && !LAPACKE_lsame( diag, 'n' ) ) ) {
57 /* Just exit if any of input parameters are wrong */
58 return (lapack_logical) 0;
59 }
60
61 if( unit ) {
62 /* Unit case, diagonal should be excluded from the check for NaN.
63 * Decoding RFP and checking both triangulars and rectangular
64 * for NaNs.
65 */
66 if( lower ) {
67 n2 = n / 2;
68 n1 = n - n2;
69 } else {
70 n1 = n / 2;
71 n2 = n - n1;
72 }
73 if( n % 2 == 1 ) {
74 /* N is odd */
75 if( ( rowmaj || ntr ) && !( rowmaj && ntr ) ) {
76 /* N is odd and ( TRANSR = 'N' .XOR. ROWMAJOR) */
77 if( lower ) {
78 return LAPACKE_str_nancheck( LAPACK_ROW_MAJOR, 'l', 'u',
79 n1, &a[0], n )
80 || LAPACKE_sge_nancheck( LAPACK_ROW_MAJOR, n2, n1,
81 &a[n1], n )
82 || LAPACKE_str_nancheck( LAPACK_ROW_MAJOR, 'u', 'u',
83 n2, &a[n], n );
84 } else {
85 return LAPACKE_str_nancheck( LAPACK_ROW_MAJOR, 'l', 'u',
86 n1, &a[n2], n )
87 || LAPACKE_sge_nancheck( LAPACK_ROW_MAJOR, n1, n2,
88 &a[0], n )
89 || LAPACKE_str_nancheck( LAPACK_ROW_MAJOR, 'u', 'u',
90 n2, &a[n1], n );
91 }
92 } else {
93 /* N is odd and
94 ( ( TRANSR = 'C' || TRANSR = 'T' ) .XOR. COLMAJOR ) */
95 if( lower ) {
96 return LAPACKE_str_nancheck( LAPACK_ROW_MAJOR, 'u', 'u',
97 n1, &a[0], n1 )
98 || LAPACKE_sge_nancheck( LAPACK_ROW_MAJOR, n1, n2,
99 &a[1], n1 )
100 || LAPACKE_str_nancheck( LAPACK_ROW_MAJOR, 'l', 'u',
101 n2, &a[1], n1 );
102 } else {
103 return LAPACKE_str_nancheck( LAPACK_ROW_MAJOR, 'u', 'u',
104 n1, &a[(size_t)n2*n2], n2 )
105 || LAPACKE_sge_nancheck( LAPACK_ROW_MAJOR, n2, n1,
106 &a[0], n2 )
107 || LAPACKE_str_nancheck( LAPACK_ROW_MAJOR, 'l', 'u',
108 n2, &a[(size_t)n1*n2], n2 );
109 }
110 }
111 } else {
112 /* N is even */
113 k = n / 2;
114 if( ( rowmaj || ntr ) && !( rowmaj && ntr ) ) {
115 /* N is even and ( TRANSR = 'N' .XOR. ROWMAJOR) */
116 if( lower ) {
117 return LAPACKE_str_nancheck( LAPACK_ROW_MAJOR, 'l', 'u',
118 k, &a[1], n+1 )
119 || LAPACKE_sge_nancheck( LAPACK_ROW_MAJOR, k, k,
120 &a[k+1], n+1 )
121 || LAPACKE_str_nancheck( LAPACK_ROW_MAJOR, 'u', 'u',
122 k, &a[0], n+1 );
123 } else {
124 return LAPACKE_str_nancheck( LAPACK_ROW_MAJOR, 'l', 'u',
125 k, &a[k+1], n+1 )
126 || LAPACKE_sge_nancheck( LAPACK_ROW_MAJOR, k, k,
127 &a[0], n+1 )
128 || LAPACKE_str_nancheck( LAPACK_ROW_MAJOR, 'u', 'u',
129 k, &a[k], n+1 );
130 }
131 } else {
132 /* N is even and
133 * ( ( TRANSR = 'C' || TRANSR = 'T' ) .XOR. COLMAJOR )
134 */
135 if( lower ) {
136 return LAPACKE_str_nancheck( LAPACK_ROW_MAJOR, 'u', 'u',
137 k, &a[k], k )
138 || LAPACKE_sge_nancheck( LAPACK_ROW_MAJOR, k, k,
139 &a[(size_t)k*(k+1)], k )
140 || LAPACKE_str_nancheck( LAPACK_ROW_MAJOR, 'l', 'u',
141 k, &a[0], k );
142 } else {
143 return LAPACKE_str_nancheck( LAPACK_ROW_MAJOR, 'u', 'u',
144 k, &a[(size_t)k*(k+1)], k )
145 || LAPACKE_sge_nancheck( LAPACK_ROW_MAJOR, k, k,
146 &a[0], k )
147 || LAPACKE_str_nancheck( LAPACK_ROW_MAJOR, 'l', 'u',
148 k, &a[(size_t)k*k], k );
149 }
150 }
151 }
152 } else {
153 /* Non-unit case - just check whole array for NaNs. */
154 len = n*(n+1)/2;
155 return LAPACKE_sge_nancheck( LAPACK_COL_MAJOR, len, 1, a, len );
156 }
157}
lapack_int
#define lapack_int
Definition: lapack.h:87
lapack_logical
#define lapack_logical
Definition: lapack.h:103
LAPACK_COL_MAJOR
#define LAPACK_COL_MAJOR
Definition: lapacke.h:53
LAPACK_ROW_MAJOR
#define LAPACK_ROW_MAJOR
Definition: lapacke.h:52
LAPACKE_lsame
lapack_logical LAPACKE_lsame(char ca, char cb)
Definition: lapacke_lsame.c:35
LAPACKE_sge_nancheck
lapack_logical LAPACKE_sge_nancheck(int matrix_layout, lapack_int m, lapack_int n, const float *a, lapack_int lda)
Definition: lapacke_sge_nancheck.c:36
LAPACKE_str_nancheck
lapack_logical LAPACKE_str_nancheck(int matrix_layout, char uplo, char diag, lapack_int n, const float *a, lapack_int lda)
Definition: lapacke_str_nancheck.c:36
Here is the call graph for this function:
Here is the caller graph for this function: