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_csytrs( char *uplo, lapack_int *n, lapack_int *nrhs,
00055 lapack_int *lda, lapack_int *ldb );
00056 static void init_a( lapack_int size, lapack_complex_float *a );
00057 static void init_ipiv( lapack_int size, lapack_int *ipiv );
00058 static void init_b( lapack_int size, lapack_complex_float *b );
00059 static int compare_csytrs( lapack_complex_float *b, lapack_complex_float *b_i,
00060 lapack_int info, lapack_int info_i, lapack_int ldb,
00061 lapack_int nrhs );
00062
00063 int main(void)
00064 {
00065
00066 char uplo, uplo_i;
00067 lapack_int n, n_i;
00068 lapack_int nrhs, nrhs_i;
00069 lapack_int lda, lda_i;
00070 lapack_int lda_r;
00071 lapack_int ldb, ldb_i;
00072 lapack_int ldb_r;
00073 lapack_int info, info_i;
00074 lapack_int i;
00075 int failed;
00076
00077
00078 lapack_complex_float *a = NULL, *a_i = NULL;
00079 lapack_int *ipiv = NULL, *ipiv_i = NULL;
00080 lapack_complex_float *b = NULL, *b_i = NULL;
00081 lapack_complex_float *b_save = NULL;
00082 lapack_complex_float *a_r = NULL;
00083 lapack_complex_float *b_r = NULL;
00084
00085
00086 init_scalars_csytrs( &uplo, &n, &nrhs, &lda, &ldb );
00087 lda_r = n+2;
00088 ldb_r = nrhs+2;
00089 uplo_i = uplo;
00090 n_i = n;
00091 nrhs_i = nrhs;
00092 lda_i = lda;
00093 ldb_i = ldb;
00094
00095
00096 a = (lapack_complex_float *)
00097 LAPACKE_malloc( lda*n * sizeof(lapack_complex_float) );
00098 ipiv = (lapack_int *)LAPACKE_malloc( n * sizeof(lapack_int) );
00099 b = (lapack_complex_float *)
00100 LAPACKE_malloc( ldb*nrhs * sizeof(lapack_complex_float) );
00101
00102
00103 a_i = (lapack_complex_float *)
00104 LAPACKE_malloc( lda*n * sizeof(lapack_complex_float) );
00105 ipiv_i = (lapack_int *)LAPACKE_malloc( n * sizeof(lapack_int) );
00106 b_i = (lapack_complex_float *)
00107 LAPACKE_malloc( ldb*nrhs * sizeof(lapack_complex_float) );
00108
00109
00110 b_save = (lapack_complex_float *)
00111 LAPACKE_malloc( ldb*nrhs * sizeof(lapack_complex_float) );
00112
00113
00114 a_r = (lapack_complex_float *)
00115 LAPACKE_malloc( n*(n+2) * sizeof(lapack_complex_float) );
00116 b_r = (lapack_complex_float *)
00117 LAPACKE_malloc( n*(nrhs+2) * sizeof(lapack_complex_float) );
00118
00119
00120 init_a( lda*n, a );
00121 init_ipiv( n, ipiv );
00122 init_b( ldb*nrhs, b );
00123
00124
00125 for( i = 0; i < ldb*nrhs; i++ ) {
00126 b_save[i] = b[i];
00127 }
00128
00129
00130 csytrs_( &uplo, &n, &nrhs, a, &lda, ipiv, b, &ldb, &info );
00131
00132
00133
00134 for( i = 0; i < lda*n; i++ ) {
00135 a_i[i] = a[i];
00136 }
00137 for( i = 0; i < n; i++ ) {
00138 ipiv_i[i] = ipiv[i];
00139 }
00140 for( i = 0; i < ldb*nrhs; i++ ) {
00141 b_i[i] = b_save[i];
00142 }
00143 info_i = LAPACKE_csytrs_work( LAPACK_COL_MAJOR, uplo_i, n_i, nrhs_i, a_i,
00144 lda_i, ipiv_i, b_i, ldb_i );
00145
00146 failed = compare_csytrs( b, b_i, info, info_i, ldb, nrhs );
00147 if( failed == 0 ) {
00148 printf( "PASSED: column-major middle-level interface to csytrs\n" );
00149 } else {
00150 printf( "FAILED: column-major middle-level interface to csytrs\n" );
00151 }
00152
00153
00154
00155 for( i = 0; i < lda*n; i++ ) {
00156 a_i[i] = a[i];
00157 }
00158 for( i = 0; i < n; i++ ) {
00159 ipiv_i[i] = ipiv[i];
00160 }
00161 for( i = 0; i < ldb*nrhs; i++ ) {
00162 b_i[i] = b_save[i];
00163 }
00164 info_i = LAPACKE_csytrs( LAPACK_COL_MAJOR, uplo_i, n_i, nrhs_i, a_i, lda_i,
00165 ipiv_i, b_i, ldb_i );
00166
00167 failed = compare_csytrs( b, b_i, info, info_i, ldb, nrhs );
00168 if( failed == 0 ) {
00169 printf( "PASSED: column-major high-level interface to csytrs\n" );
00170 } else {
00171 printf( "FAILED: column-major high-level interface to csytrs\n" );
00172 }
00173
00174
00175
00176 for( i = 0; i < lda*n; i++ ) {
00177 a_i[i] = a[i];
00178 }
00179 for( i = 0; i < n; i++ ) {
00180 ipiv_i[i] = ipiv[i];
00181 }
00182 for( i = 0; i < ldb*nrhs; i++ ) {
00183 b_i[i] = b_save[i];
00184 }
00185
00186 LAPACKE_cge_trans( LAPACK_COL_MAJOR, n, n, a_i, lda, a_r, n+2 );
00187 LAPACKE_cge_trans( LAPACK_COL_MAJOR, n, nrhs, b_i, ldb, b_r, nrhs+2 );
00188 info_i = LAPACKE_csytrs_work( LAPACK_ROW_MAJOR, uplo_i, n_i, nrhs_i, a_r,
00189 lda_r, ipiv_i, b_r, ldb_r );
00190
00191 LAPACKE_cge_trans( LAPACK_ROW_MAJOR, n, nrhs, b_r, nrhs+2, b_i, ldb );
00192
00193 failed = compare_csytrs( b, b_i, info, info_i, ldb, nrhs );
00194 if( failed == 0 ) {
00195 printf( "PASSED: row-major middle-level interface to csytrs\n" );
00196 } else {
00197 printf( "FAILED: row-major middle-level interface to csytrs\n" );
00198 }
00199
00200
00201
00202 for( i = 0; i < lda*n; i++ ) {
00203 a_i[i] = a[i];
00204 }
00205 for( i = 0; i < n; i++ ) {
00206 ipiv_i[i] = ipiv[i];
00207 }
00208 for( i = 0; i < ldb*nrhs; i++ ) {
00209 b_i[i] = b_save[i];
00210 }
00211
00212
00213 LAPACKE_cge_trans( LAPACK_COL_MAJOR, n, n, a_i, lda, a_r, n+2 );
00214 LAPACKE_cge_trans( LAPACK_COL_MAJOR, n, nrhs, b_i, ldb, b_r, nrhs+2 );
00215 info_i = LAPACKE_csytrs( LAPACK_ROW_MAJOR, uplo_i, n_i, nrhs_i, a_r, lda_r,
00216 ipiv_i, b_r, ldb_r );
00217
00218 LAPACKE_cge_trans( LAPACK_ROW_MAJOR, n, nrhs, b_r, nrhs+2, b_i, ldb );
00219
00220 failed = compare_csytrs( b, b_i, info, info_i, ldb, nrhs );
00221 if( failed == 0 ) {
00222 printf( "PASSED: row-major high-level interface to csytrs\n" );
00223 } else {
00224 printf( "FAILED: row-major high-level interface to csytrs\n" );
00225 }
00226
00227
00228 if( a != NULL ) {
00229 LAPACKE_free( a );
00230 }
00231 if( a_i != NULL ) {
00232 LAPACKE_free( a_i );
00233 }
00234 if( a_r != NULL ) {
00235 LAPACKE_free( a_r );
00236 }
00237 if( ipiv != NULL ) {
00238 LAPACKE_free( ipiv );
00239 }
00240 if( ipiv_i != NULL ) {
00241 LAPACKE_free( ipiv_i );
00242 }
00243 if( b != NULL ) {
00244 LAPACKE_free( b );
00245 }
00246 if( b_i != NULL ) {
00247 LAPACKE_free( b_i );
00248 }
00249 if( b_r != NULL ) {
00250 LAPACKE_free( b_r );
00251 }
00252 if( b_save != NULL ) {
00253 LAPACKE_free( b_save );
00254 }
00255
00256 return 0;
00257 }
00258
00259
00260 static void init_scalars_csytrs( char *uplo, lapack_int *n, lapack_int *nrhs,
00261 lapack_int *lda, lapack_int *ldb )
00262 {
00263 *uplo = 'L';
00264 *n = 4;
00265 *nrhs = 2;
00266 *lda = 8;
00267 *ldb = 8;
00268
00269 return;
00270 }
00271
00272
00273 static void init_a( lapack_int size, lapack_complex_float *a ) {
00274 lapack_int i;
00275 for( i = 0; i < size; i++ ) {
00276 a[i] = lapack_make_complex_float( 0.0f, 0.0f );
00277 }
00278 a[0] = lapack_make_complex_float( -3.899999857e-001, -7.099999785e-001 );
00279 a[8] = lapack_make_complex_float( 0.000000000e+000, 0.000000000e+000 );
00280 a[16] = lapack_make_complex_float( 0.000000000e+000, 0.000000000e+000 );
00281 a[24] = lapack_make_complex_float( 0.000000000e+000, 0.000000000e+000 );
00282 a[1] = lapack_make_complex_float( -7.860000134e+000, -2.960000038e+000 );
00283 a[9] = lapack_make_complex_float( -2.829999924e+000, -2.999999933e-002 );
00284 a[17] = lapack_make_complex_float( 0.000000000e+000, 0.000000000e+000 );
00285 a[25] = lapack_make_complex_float( 0.000000000e+000, 0.000000000e+000 );
00286 a[2] = lapack_make_complex_float( 5.278724432e-001, -3.714659810e-001 );
00287 a[10] = lapack_make_complex_float( -6.078390479e-001, 2.810796499e-001 );
00288 a[18] = lapack_make_complex_float( 4.407906532e+000, 5.399120331e+000 );
00289 a[26] = lapack_make_complex_float( 0.000000000e+000, 0.000000000e+000 );
00290 a[3] = lapack_make_complex_float( 4.425582290e-001, 1.936483681e-001 );
00291 a[11] = lapack_make_complex_float( -4.822822809e-001, 1.498936117e-002 );
00292 a[19] = lapack_make_complex_float( -1.070821062e-001, -3.156781197e-001 );
00293 a[27] = lapack_make_complex_float( -2.095414639e+000, -2.201138735e+000 );
00294 }
00295 static void init_ipiv( lapack_int size, lapack_int *ipiv ) {
00296 lapack_int i;
00297 for( i = 0; i < size; i++ ) {
00298 ipiv[i] = 0;
00299 }
00300 ipiv[0] = -3;
00301 ipiv[1] = -3;
00302 ipiv[2] = 3;
00303 ipiv[3] = 4;
00304 }
00305 static void init_b( lapack_int size, lapack_complex_float *b ) {
00306 lapack_int i;
00307 for( i = 0; i < size; i++ ) {
00308 b[i] = lapack_make_complex_float( 0.0f, 0.0f );
00309 }
00310 b[0] = lapack_make_complex_float( -5.563999939e+001, 4.122000122e+001 );
00311 b[8] = lapack_make_complex_float( -1.909000015e+001, -3.597000122e+001 );
00312 b[1] = lapack_make_complex_float( -4.818000031e+001, 6.600000000e+001 );
00313 b[9] = lapack_make_complex_float( -1.207999992e+001, -2.702000046e+001 );
00314 b[2] = lapack_make_complex_float( -4.900000095e-001, -1.470000029e+000 );
00315 b[10] = lapack_make_complex_float( 6.949999809e+000, 2.048999977e+001 );
00316 b[3] = lapack_make_complex_float( -6.429999828e+000, 1.923999977e+001 );
00317 b[11] = lapack_make_complex_float( -4.590000153e+000, -3.552999878e+001 );
00318 }
00319
00320
00321
00322 static int compare_csytrs( lapack_complex_float *b, lapack_complex_float *b_i,
00323 lapack_int info, lapack_int info_i, lapack_int ldb,
00324 lapack_int nrhs )
00325 {
00326 lapack_int i;
00327 int failed = 0;
00328 for( i = 0; i < ldb*nrhs; i++ ) {
00329 failed += compare_complex_floats(b[i],b_i[i]);
00330 }
00331 failed += (info == info_i) ? 0 : 1;
00332 if( info != 0 || info_i != 0 ) {
00333 printf( "info=%d, info_i=%d\n",(int)info,(int)info_i );
00334 }
00335
00336 return failed;
00337 }