00001
00002
00003
00004
00005
00006
00007
00008
00009
00010
00011
00012
00013
00014
00015
00016
00017
00018
00019
00020
00021
00022
00023
00024
00025
00026
00027
00028
00029
00030
00031
00032
00033
00034
00035
00036
00037
00038
00039
00040
00041
00042
00043
00044
00045
00046
00047
00048
00049 #include <stdio.h>
00050 #include "lapacke.h"
00051 #include "lapacke_utils.h"
00052 #include "test_utils.h"
00053
00054 static void init_scalars_zpbtrf( char *uplo, lapack_int *n, lapack_int *kd,
00055 lapack_int *ldab );
00056 static void init_ab( lapack_int size, lapack_complex_double *ab );
00057 static int compare_zpbtrf( lapack_complex_double *ab,
00058 lapack_complex_double *ab_i, lapack_int info,
00059 lapack_int info_i, lapack_int ldab, lapack_int n );
00060
00061 int main(void)
00062 {
00063
00064 char uplo, uplo_i;
00065 lapack_int n, n_i;
00066 lapack_int kd, kd_i;
00067 lapack_int ldab, ldab_i;
00068 lapack_int ldab_r;
00069 lapack_int info, info_i;
00070 lapack_int i;
00071 int failed;
00072
00073
00074 lapack_complex_double *ab = NULL, *ab_i = NULL;
00075 lapack_complex_double *ab_save = NULL;
00076 lapack_complex_double *ab_r = NULL;
00077
00078
00079 init_scalars_zpbtrf( &uplo, &n, &kd, &ldab );
00080 ldab_r = n+2;
00081 uplo_i = uplo;
00082 n_i = n;
00083 kd_i = kd;
00084 ldab_i = ldab;
00085
00086
00087 ab = (lapack_complex_double *)
00088 LAPACKE_malloc( ldab*n * sizeof(lapack_complex_double) );
00089
00090
00091 ab_i = (lapack_complex_double *)
00092 LAPACKE_malloc( ldab*n * sizeof(lapack_complex_double) );
00093
00094
00095 ab_save = (lapack_complex_double *)
00096 LAPACKE_malloc( ldab*n * sizeof(lapack_complex_double) );
00097
00098
00099 ab_r = (lapack_complex_double *)
00100 LAPACKE_malloc( (kd+1)*(n+2) * sizeof(lapack_complex_double) );
00101
00102
00103 init_ab( ldab*n, ab );
00104
00105
00106 for( i = 0; i < ldab*n; i++ ) {
00107 ab_save[i] = ab[i];
00108 }
00109
00110
00111 zpbtrf_( &uplo, &n, &kd, ab, &ldab, &info );
00112
00113
00114
00115 for( i = 0; i < ldab*n; i++ ) {
00116 ab_i[i] = ab_save[i];
00117 }
00118 info_i = LAPACKE_zpbtrf_work( LAPACK_COL_MAJOR, uplo_i, n_i, kd_i, ab_i,
00119 ldab_i );
00120
00121 failed = compare_zpbtrf( ab, ab_i, info, info_i, ldab, n );
00122 if( failed == 0 ) {
00123 printf( "PASSED: column-major middle-level interface to zpbtrf\n" );
00124 } else {
00125 printf( "FAILED: column-major middle-level interface to zpbtrf\n" );
00126 }
00127
00128
00129
00130 for( i = 0; i < ldab*n; i++ ) {
00131 ab_i[i] = ab_save[i];
00132 }
00133 info_i = LAPACKE_zpbtrf( LAPACK_COL_MAJOR, uplo_i, n_i, kd_i, ab_i,
00134 ldab_i );
00135
00136 failed = compare_zpbtrf( ab, ab_i, info, info_i, ldab, n );
00137 if( failed == 0 ) {
00138 printf( "PASSED: column-major high-level interface to zpbtrf\n" );
00139 } else {
00140 printf( "FAILED: column-major high-level interface to zpbtrf\n" );
00141 }
00142
00143
00144
00145 for( i = 0; i < ldab*n; i++ ) {
00146 ab_i[i] = ab_save[i];
00147 }
00148
00149 LAPACKE_zge_trans( LAPACK_COL_MAJOR, kd+1, n, ab_i, ldab, ab_r, n+2 );
00150 info_i = LAPACKE_zpbtrf_work( LAPACK_ROW_MAJOR, uplo_i, n_i, kd_i, ab_r,
00151 ldab_r );
00152
00153 LAPACKE_zge_trans( LAPACK_ROW_MAJOR, kd+1, n, ab_r, n+2, ab_i, ldab );
00154
00155 failed = compare_zpbtrf( ab, ab_i, info, info_i, ldab, n );
00156 if( failed == 0 ) {
00157 printf( "PASSED: row-major middle-level interface to zpbtrf\n" );
00158 } else {
00159 printf( "FAILED: row-major middle-level interface to zpbtrf\n" );
00160 }
00161
00162
00163
00164 for( i = 0; i < ldab*n; i++ ) {
00165 ab_i[i] = ab_save[i];
00166 }
00167
00168
00169 LAPACKE_zge_trans( LAPACK_COL_MAJOR, kd+1, n, ab_i, ldab, ab_r, n+2 );
00170 info_i = LAPACKE_zpbtrf( LAPACK_ROW_MAJOR, uplo_i, n_i, kd_i, ab_r,
00171 ldab_r );
00172
00173 LAPACKE_zge_trans( LAPACK_ROW_MAJOR, kd+1, n, ab_r, n+2, ab_i, ldab );
00174
00175 failed = compare_zpbtrf( ab, ab_i, info, info_i, ldab, n );
00176 if( failed == 0 ) {
00177 printf( "PASSED: row-major high-level interface to zpbtrf\n" );
00178 } else {
00179 printf( "FAILED: row-major high-level interface to zpbtrf\n" );
00180 }
00181
00182
00183 if( ab != NULL ) {
00184 LAPACKE_free( ab );
00185 }
00186 if( ab_i != NULL ) {
00187 LAPACKE_free( ab_i );
00188 }
00189 if( ab_r != NULL ) {
00190 LAPACKE_free( ab_r );
00191 }
00192 if( ab_save != NULL ) {
00193 LAPACKE_free( ab_save );
00194 }
00195
00196 return 0;
00197 }
00198
00199
00200 static void init_scalars_zpbtrf( char *uplo, lapack_int *n, lapack_int *kd,
00201 lapack_int *ldab )
00202 {
00203 *uplo = 'L';
00204 *n = 4;
00205 *kd = 1;
00206 *ldab = 9;
00207
00208 return;
00209 }
00210
00211
00212 static void init_ab( lapack_int size, lapack_complex_double *ab ) {
00213 lapack_int i;
00214 for( i = 0; i < size; i++ ) {
00215 ab[i] = lapack_make_complex_double( 0.0, 0.0 );
00216 }
00217 ab[0] = lapack_make_complex_double( 9.39000000000000060e+000,
00218 0.00000000000000000e+000 );
00219 ab[9] = lapack_make_complex_double( 1.68999999999999990e+000,
00220 0.00000000000000000e+000 );
00221 ab[18] = lapack_make_complex_double( 2.64999999999999990e+000,
00222 0.00000000000000000e+000 );
00223 ab[27] = lapack_make_complex_double( 2.16999999999999990e+000,
00224 0.00000000000000000e+000 );
00225 ab[1] = lapack_make_complex_double( 1.08000000000000010e+000,
00226 1.73000000000000000e+000 );
00227 ab[10] = lapack_make_complex_double( -4.00000000000000010e-002,
00228 -2.89999999999999980e-001 );
00229 ab[19] = lapack_make_complex_double( -3.30000000000000020e-001,
00230 -2.24000000000000020e+000 );
00231 ab[28] = lapack_make_complex_double( 0.00000000000000000e+000,
00232 0.00000000000000000e+000 );
00233 }
00234
00235
00236
00237 static int compare_zpbtrf( lapack_complex_double *ab,
00238 lapack_complex_double *ab_i, lapack_int info,
00239 lapack_int info_i, lapack_int ldab, lapack_int n )
00240 {
00241 lapack_int i;
00242 int failed = 0;
00243 for( i = 0; i < ldab*n; i++ ) {
00244 failed += compare_complex_doubles(ab[i],ab_i[i]);
00245 }
00246 failed += (info == info_i) ? 0 : 1;
00247 if( info != 0 || info_i != 0 ) {
00248 printf( "info=%d, info_i=%d\n",(int)info,(int)info_i );
00249 }
00250
00251 return failed;
00252 }