1 files changed, 85 insertions, 0 deletions
diff --git a/lapacke/utils/lapacke_ztr_nancheck.c b/lapacke/utils/lapacke_ztr_nancheck.c
new file mode 100644
index 00000000..291ca90b
--- /dev/null
+++ b/lapacke/utils/lapacke_ztr_nancheck.c
@@ -0,0 +1,85 @@
+/*****************************************************************************
+  Copyright (c) 2010, Intel Corp.
+  All rights reserved.
+
+  Redistribution and use in source and binary forms, with or without
+  modification, are permitted provided that the following conditions are met:
+
+    * Redistributions of source code must retain the above copyright notice,
+      this list of conditions and the following disclaimer.
+    * Redistributions in binary form must reproduce the above copyright
+      notice, this list of conditions and the following disclaimer in the
+      documentation and/or other materials provided with the distribution.
+    * Neither the name of Intel Corporation nor the names of its contributors
+      may be used to endorse or promote products derived from this software
+      without specific prior written permission.
+
+  THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
+  AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
+  IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
+  ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE
+  LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
+  CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
+  SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
+  INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
+  CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
+  ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF
+  THE POSSIBILITY OF SUCH DAMAGE.
+******************************************************************************
+* Contents: Native C interface to LAPACK utility function
+* Author: Intel Corporation
+* Created in February, 2010
+*****************************************************************************/
+#include "lapacke_utils.h"
+
+/* Check a matrix for NaN entries. */
+
+lapack_logical LAPACKE_ztr_nancheck( int matrix_order, char uplo, char diag,
+                                      lapack_int n,
+                                      const lapack_complex_double *a,
+                                      lapack_int lda )
+{
+    lapack_int i, j, st;
+    lapack_logical colmaj, lower, unit;
+
+    if( a == NULL ) return (lapack_logical) 0;
+
+    colmaj = ( matrix_order == LAPACK_COL_MAJOR );
+    lower  = LAPACKE_lsame( uplo, 'l' );
+    unit   = LAPACKE_lsame( diag, 'u' );
+
+    if( ( !colmaj && ( matrix_order != LAPACK_ROW_MAJOR ) ) ||
+        ( !lower  && !LAPACKE_lsame( uplo, 'u' ) ) ||
+        ( !unit   && !LAPACKE_lsame( diag, 'n' ) ) ) {
+        /* Just exit if any of input parameters are wrong */
+        return (lapack_logical) 0;
+    }
+    if( unit ) {
+        /* If unit, then don't touch diagonal, start from 1st column or row */
+        st = 1;
+    } else  {
+        /* If non-unit, then check diagonal also, starting from [0,0] */
+        st = 0;
+    }
+
+    /* Since col_major upper and row_major lower are equal,
+     * and col_major lower and row_major upper are equals too -
+     * using one code for equal cases. XOR( colmaj, upper )
+     */
+    if( ( colmaj || lower ) && !( colmaj && lower ) ) {
+        for( j = st; j < n; j++ ) {
+            for( i = 0; i < MIN( j+1-st, lda ); i++ ) {
+                if( LAPACK_ZISNAN( a[i+j*lda] ) )
+                    return (lapack_logical) 1;
+            }
+        }
+    } else {
+        for( j = 0; j < n-st; j++ ) {
+            for( i = j+st; i < MIN( n, lda ); i++ ) {
+                if( LAPACK_ZISNAN( a[i+j*lda] ) )
+                    return (lapack_logical) 1;
+            }
+        }
+    }
+    return (lapack_logical) 0;
+}