ctrrfs_1.c
Go to the documentation of this file.
00001 /*****************************************************************************
00002   Copyright (c) 2010, Intel Corp.
00003   All rights reserved.
00004 
00005   Redistribution and use in source and binary forms, with or without
00006   modification, are permitted provided that the following conditions are met:
00007 
00008     * Redistributions of source code must retain the above copyright notice,
00009       this list of conditions and the following disclaimer.
00010     * Redistributions in binary form must reproduce the above copyright
00011       notice, this list of conditions and the following disclaimer in the
00012       documentation and/or other materials provided with the distribution.
00013     * Neither the name of Intel Corporation nor the names of its contributors
00014       may be used to endorse or promote products derived from this software
00015       without specific prior written permission.
00016 
00017   THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
00018   AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
00019   IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
00020   ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE
00021   LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
00022   CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
00023   SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
00024   INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
00025   CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
00026   ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF
00027   THE POSSIBILITY OF SUCH DAMAGE.
00028 *****************************************************************************/
00029 /*  Contents: test routine for C interface to LAPACK
00030 *   Author: Intel Corporation
00031 *   Created in March, 2010
00032 *
00033 * Purpose
00034 *
00035 * ctrrfs_1 is the test program for the C interface to LAPACK
00036 * routine ctrrfs
00037 * The program doesn't require an input, the input data is hardcoded in the
00038 * test program.
00039 * The program tests the C interface in the four combinations:
00040 *   1) column-major layout, middle-level interface
00041 *   2) column-major layout, high-level interface
00042 *   3) row-major layout, middle-level interface
00043 *   4) row-major layout, high-level interface
00044 * The output of the C interface function is compared to those obtained from
00045 * the corresponiding LAPACK routine with the same input data, and the
00046 * comparison diagnostics is then printed on the standard output having PASSED
00047 * keyword if the test is passed, and FAILED keyword if the test isn't passed.
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_ctrrfs( char *uplo, char *trans, char *diag,
00055                                  lapack_int *n, lapack_int *nrhs,
00056                                  lapack_int *lda, lapack_int *ldb,
00057                                  lapack_int *ldx );
00058 static void init_a( lapack_int size, lapack_complex_float *a );
00059 static void init_b( lapack_int size, lapack_complex_float *b );
00060 static void init_x( lapack_int size, lapack_complex_float *x );
00061 static void init_ferr( lapack_int size, float *ferr );
00062 static void init_berr( lapack_int size, float *berr );
00063 static void init_work( lapack_int size, lapack_complex_float *work );
00064 static void init_rwork( lapack_int size, float *rwork );
00065 static int compare_ctrrfs( float *ferr, float *ferr_i, float *berr,
00066                            float *berr_i, lapack_int info, lapack_int info_i,
00067                            lapack_int nrhs );
00068 
00069 int main(void)
00070 {
00071     /* Local scalars */
00072     char uplo, uplo_i;
00073     char trans, trans_i;
00074     char diag, diag_i;
00075     lapack_int n, n_i;
00076     lapack_int nrhs, nrhs_i;
00077     lapack_int lda, lda_i;
00078     lapack_int lda_r;
00079     lapack_int ldb, ldb_i;
00080     lapack_int ldb_r;
00081     lapack_int ldx, ldx_i;
00082     lapack_int ldx_r;
00083     lapack_int info, info_i;
00084     lapack_int i;
00085     int failed;
00086 
00087     /* Local arrays */
00088     lapack_complex_float *a = NULL, *a_i = NULL;
00089     lapack_complex_float *b = NULL, *b_i = NULL;
00090     lapack_complex_float *x = NULL, *x_i = NULL;
00091     float *ferr = NULL, *ferr_i = NULL;
00092     float *berr = NULL, *berr_i = NULL;
00093     lapack_complex_float *work = NULL, *work_i = NULL;
00094     float *rwork = NULL, *rwork_i = NULL;
00095     float *ferr_save = NULL;
00096     float *berr_save = NULL;
00097     lapack_complex_float *a_r = NULL;
00098     lapack_complex_float *b_r = NULL;
00099     lapack_complex_float *x_r = NULL;
00100 
00101     /* Iniitialize the scalar parameters */
00102     init_scalars_ctrrfs( &uplo, &trans, &diag, &n, &nrhs, &lda, &ldb, &ldx );
00103     lda_r = n+2;
00104     ldb_r = nrhs+2;
00105     ldx_r = nrhs+2;
00106     uplo_i = uplo;
00107     trans_i = trans;
00108     diag_i = diag;
00109     n_i = n;
00110     nrhs_i = nrhs;
00111     lda_i = lda;
00112     ldb_i = ldb;
00113     ldx_i = ldx;
00114 
00115     /* Allocate memory for the LAPACK routine arrays */
00116     a = (lapack_complex_float *)
00117         LAPACKE_malloc( lda*n * sizeof(lapack_complex_float) );
00118     b = (lapack_complex_float *)
00119         LAPACKE_malloc( ldb*nrhs * sizeof(lapack_complex_float) );
00120     x = (lapack_complex_float *)
00121         LAPACKE_malloc( ldx*nrhs * sizeof(lapack_complex_float) );
00122     ferr = (float *)LAPACKE_malloc( nrhs * sizeof(float) );
00123     berr = (float *)LAPACKE_malloc( nrhs * sizeof(float) );
00124     work = (lapack_complex_float *)
00125         LAPACKE_malloc( 2*n * sizeof(lapack_complex_float) );
00126     rwork = (float *)LAPACKE_malloc( n * sizeof(float) );
00127 
00128     /* Allocate memory for the C interface function arrays */
00129     a_i = (lapack_complex_float *)
00130         LAPACKE_malloc( lda*n * sizeof(lapack_complex_float) );
00131     b_i = (lapack_complex_float *)
00132         LAPACKE_malloc( ldb*nrhs * sizeof(lapack_complex_float) );
00133     x_i = (lapack_complex_float *)
00134         LAPACKE_malloc( ldx*nrhs * sizeof(lapack_complex_float) );
00135     ferr_i = (float *)LAPACKE_malloc( nrhs * sizeof(float) );
00136     berr_i = (float *)LAPACKE_malloc( nrhs * sizeof(float) );
00137     work_i = (lapack_complex_float *)
00138         LAPACKE_malloc( 2*n * sizeof(lapack_complex_float) );
00139     rwork_i = (float *)LAPACKE_malloc( n * sizeof(float) );
00140 
00141     /* Allocate memory for the backup arrays */
00142     ferr_save = (float *)LAPACKE_malloc( nrhs * sizeof(float) );
00143     berr_save = (float *)LAPACKE_malloc( nrhs * sizeof(float) );
00144 
00145     /* Allocate memory for the row-major arrays */
00146     a_r = (lapack_complex_float *)
00147         LAPACKE_malloc( n*(n+2) * sizeof(lapack_complex_float) );
00148     b_r = (lapack_complex_float *)
00149         LAPACKE_malloc( n*(nrhs+2) * sizeof(lapack_complex_float) );
00150     x_r = (lapack_complex_float *)
00151         LAPACKE_malloc( n*(nrhs+2) * sizeof(lapack_complex_float) );
00152 
00153     /* Initialize input arrays */
00154     init_a( lda*n, a );
00155     init_b( ldb*nrhs, b );
00156     init_x( ldx*nrhs, x );
00157     init_ferr( nrhs, ferr );
00158     init_berr( nrhs, berr );
00159     init_work( 2*n, work );
00160     init_rwork( n, rwork );
00161 
00162     /* Backup the ouptut arrays */
00163     for( i = 0; i < nrhs; i++ ) {
00164         ferr_save[i] = ferr[i];
00165     }
00166     for( i = 0; i < nrhs; i++ ) {
00167         berr_save[i] = berr[i];
00168     }
00169 
00170     /* Call the LAPACK routine */
00171     ctrrfs_( &uplo, &trans, &diag, &n, &nrhs, a, &lda, b, &ldb, x, &ldx, ferr,
00172              berr, work, rwork, &info );
00173 
00174     /* Initialize input data, call the column-major middle-level
00175      * interface to LAPACK routine and check the results */
00176     for( i = 0; i < lda*n; i++ ) {
00177         a_i[i] = a[i];
00178     }
00179     for( i = 0; i < ldb*nrhs; i++ ) {
00180         b_i[i] = b[i];
00181     }
00182     for( i = 0; i < ldx*nrhs; i++ ) {
00183         x_i[i] = x[i];
00184     }
00185     for( i = 0; i < nrhs; i++ ) {
00186         ferr_i[i] = ferr_save[i];
00187     }
00188     for( i = 0; i < nrhs; i++ ) {
00189         berr_i[i] = berr_save[i];
00190     }
00191     for( i = 0; i < 2*n; i++ ) {
00192         work_i[i] = work[i];
00193     }
00194     for( i = 0; i < n; i++ ) {
00195         rwork_i[i] = rwork[i];
00196     }
00197     info_i = LAPACKE_ctrrfs_work( LAPACK_COL_MAJOR, uplo_i, trans_i, diag_i,
00198                                   n_i, nrhs_i, a_i, lda_i, b_i, ldb_i, x_i,
00199                                   ldx_i, ferr_i, berr_i, work_i, rwork_i );
00200 
00201     failed = compare_ctrrfs( ferr, ferr_i, berr, berr_i, info, info_i, nrhs );
00202     if( failed == 0 ) {
00203         printf( "PASSED: column-major middle-level interface to ctrrfs\n" );
00204     } else {
00205         printf( "FAILED: column-major middle-level interface to ctrrfs\n" );
00206     }
00207 
00208     /* Initialize input data, call the column-major high-level
00209      * interface to LAPACK routine and check the results */
00210     for( i = 0; i < lda*n; i++ ) {
00211         a_i[i] = a[i];
00212     }
00213     for( i = 0; i < ldb*nrhs; i++ ) {
00214         b_i[i] = b[i];
00215     }
00216     for( i = 0; i < ldx*nrhs; i++ ) {
00217         x_i[i] = x[i];
00218     }
00219     for( i = 0; i < nrhs; i++ ) {
00220         ferr_i[i] = ferr_save[i];
00221     }
00222     for( i = 0; i < nrhs; i++ ) {
00223         berr_i[i] = berr_save[i];
00224     }
00225     for( i = 0; i < 2*n; i++ ) {
00226         work_i[i] = work[i];
00227     }
00228     for( i = 0; i < n; i++ ) {
00229         rwork_i[i] = rwork[i];
00230     }
00231     info_i = LAPACKE_ctrrfs( LAPACK_COL_MAJOR, uplo_i, trans_i, diag_i, n_i,
00232                              nrhs_i, a_i, lda_i, b_i, ldb_i, x_i, ldx_i, ferr_i,
00233                              berr_i );
00234 
00235     failed = compare_ctrrfs( ferr, ferr_i, berr, berr_i, info, info_i, nrhs );
00236     if( failed == 0 ) {
00237         printf( "PASSED: column-major high-level interface to ctrrfs\n" );
00238     } else {
00239         printf( "FAILED: column-major high-level interface to ctrrfs\n" );
00240     }
00241 
00242     /* Initialize input data, call the row-major middle-level
00243      * interface to LAPACK routine and check the results */
00244     for( i = 0; i < lda*n; i++ ) {
00245         a_i[i] = a[i];
00246     }
00247     for( i = 0; i < ldb*nrhs; i++ ) {
00248         b_i[i] = b[i];
00249     }
00250     for( i = 0; i < ldx*nrhs; i++ ) {
00251         x_i[i] = x[i];
00252     }
00253     for( i = 0; i < nrhs; i++ ) {
00254         ferr_i[i] = ferr_save[i];
00255     }
00256     for( i = 0; i < nrhs; i++ ) {
00257         berr_i[i] = berr_save[i];
00258     }
00259     for( i = 0; i < 2*n; i++ ) {
00260         work_i[i] = work[i];
00261     }
00262     for( i = 0; i < n; i++ ) {
00263         rwork_i[i] = rwork[i];
00264     }
00265 
00266     LAPACKE_cge_trans( LAPACK_COL_MAJOR, n, n, a_i, lda, a_r, n+2 );
00267     LAPACKE_cge_trans( LAPACK_COL_MAJOR, n, nrhs, b_i, ldb, b_r, nrhs+2 );
00268     LAPACKE_cge_trans( LAPACK_COL_MAJOR, n, nrhs, x_i, ldx, x_r, nrhs+2 );
00269     info_i = LAPACKE_ctrrfs_work( LAPACK_ROW_MAJOR, uplo_i, trans_i, diag_i,
00270                                   n_i, nrhs_i, a_r, lda_r, b_r, ldb_r, x_r,
00271                                   ldx_r, ferr_i, berr_i, work_i, rwork_i );
00272 
00273     failed = compare_ctrrfs( ferr, ferr_i, berr, berr_i, info, info_i, nrhs );
00274     if( failed == 0 ) {
00275         printf( "PASSED: row-major middle-level interface to ctrrfs\n" );
00276     } else {
00277         printf( "FAILED: row-major middle-level interface to ctrrfs\n" );
00278     }
00279 
00280     /* Initialize input data, call the row-major high-level
00281      * interface to LAPACK routine and check the results */
00282     for( i = 0; i < lda*n; i++ ) {
00283         a_i[i] = a[i];
00284     }
00285     for( i = 0; i < ldb*nrhs; i++ ) {
00286         b_i[i] = b[i];
00287     }
00288     for( i = 0; i < ldx*nrhs; i++ ) {
00289         x_i[i] = x[i];
00290     }
00291     for( i = 0; i < nrhs; i++ ) {
00292         ferr_i[i] = ferr_save[i];
00293     }
00294     for( i = 0; i < nrhs; i++ ) {
00295         berr_i[i] = berr_save[i];
00296     }
00297     for( i = 0; i < 2*n; i++ ) {
00298         work_i[i] = work[i];
00299     }
00300     for( i = 0; i < n; i++ ) {
00301         rwork_i[i] = rwork[i];
00302     }
00303 
00304     /* Init row_major arrays */
00305     LAPACKE_cge_trans( LAPACK_COL_MAJOR, n, n, a_i, lda, a_r, n+2 );
00306     LAPACKE_cge_trans( LAPACK_COL_MAJOR, n, nrhs, b_i, ldb, b_r, nrhs+2 );
00307     LAPACKE_cge_trans( LAPACK_COL_MAJOR, n, nrhs, x_i, ldx, x_r, nrhs+2 );
00308     info_i = LAPACKE_ctrrfs( LAPACK_ROW_MAJOR, uplo_i, trans_i, diag_i, n_i,
00309                              nrhs_i, a_r, lda_r, b_r, ldb_r, x_r, ldx_r, ferr_i,
00310                              berr_i );
00311 
00312     failed = compare_ctrrfs( ferr, ferr_i, berr, berr_i, info, info_i, nrhs );
00313     if( failed == 0 ) {
00314         printf( "PASSED: row-major high-level interface to ctrrfs\n" );
00315     } else {
00316         printf( "FAILED: row-major high-level interface to ctrrfs\n" );
00317     }
00318 
00319     /* Release memory */
00320     if( a != NULL ) {
00321         LAPACKE_free( a );
00322     }
00323     if( a_i != NULL ) {
00324         LAPACKE_free( a_i );
00325     }
00326     if( a_r != NULL ) {
00327         LAPACKE_free( a_r );
00328     }
00329     if( b != NULL ) {
00330         LAPACKE_free( b );
00331     }
00332     if( b_i != NULL ) {
00333         LAPACKE_free( b_i );
00334     }
00335     if( b_r != NULL ) {
00336         LAPACKE_free( b_r );
00337     }
00338     if( x != NULL ) {
00339         LAPACKE_free( x );
00340     }
00341     if( x_i != NULL ) {
00342         LAPACKE_free( x_i );
00343     }
00344     if( x_r != NULL ) {
00345         LAPACKE_free( x_r );
00346     }
00347     if( ferr != NULL ) {
00348         LAPACKE_free( ferr );
00349     }
00350     if( ferr_i != NULL ) {
00351         LAPACKE_free( ferr_i );
00352     }
00353     if( ferr_save != NULL ) {
00354         LAPACKE_free( ferr_save );
00355     }
00356     if( berr != NULL ) {
00357         LAPACKE_free( berr );
00358     }
00359     if( berr_i != NULL ) {
00360         LAPACKE_free( berr_i );
00361     }
00362     if( berr_save != NULL ) {
00363         LAPACKE_free( berr_save );
00364     }
00365     if( work != NULL ) {
00366         LAPACKE_free( work );
00367     }
00368     if( work_i != NULL ) {
00369         LAPACKE_free( work_i );
00370     }
00371     if( rwork != NULL ) {
00372         LAPACKE_free( rwork );
00373     }
00374     if( rwork_i != NULL ) {
00375         LAPACKE_free( rwork_i );
00376     }
00377 
00378     return 0;
00379 }
00380 
00381 /* Auxiliary function: ctrrfs scalar parameters initialization */
00382 static void init_scalars_ctrrfs( char *uplo, char *trans, char *diag,
00383                                  lapack_int *n, lapack_int *nrhs,
00384                                  lapack_int *lda, lapack_int *ldb,
00385                                  lapack_int *ldx )
00386 {
00387     *uplo = 'L';
00388     *trans = 'N';
00389     *diag = 'N';
00390     *n = 4;
00391     *nrhs = 2;
00392     *lda = 8;
00393     *ldb = 8;
00394     *ldx = 8;
00395 
00396     return;
00397 }
00398 
00399 /* Auxiliary functions: ctrrfs array parameters initialization */
00400 static void init_a( lapack_int size, lapack_complex_float *a ) {
00401     lapack_int i;
00402     for( i = 0; i < size; i++ ) {
00403         a[i] = lapack_make_complex_float( 0.0f, 0.0f );
00404     }
00405     a[0] = lapack_make_complex_float( 4.780000210e+000, 4.559999943e+000 );
00406     a[8] = lapack_make_complex_float( 0.000000000e+000, 0.000000000e+000 );
00407     a[16] = lapack_make_complex_float( 0.000000000e+000, 0.000000000e+000 );
00408     a[24] = lapack_make_complex_float( 0.000000000e+000, 0.000000000e+000 );
00409     a[1] = lapack_make_complex_float( 2.000000000e+000, -3.000000119e-001 );
00410     a[9] = lapack_make_complex_float( -4.110000134e+000, 1.250000000e+000 );
00411     a[17] = lapack_make_complex_float( 0.000000000e+000, 0.000000000e+000 );
00412     a[25] = lapack_make_complex_float( 0.000000000e+000, 0.000000000e+000 );
00413     a[2] = lapack_make_complex_float( 2.890000105e+000, -1.340000033e+000 );
00414     a[10] = lapack_make_complex_float( 2.359999895e+000, -4.250000000e+000 );
00415     a[18] = lapack_make_complex_float( 4.150000095e+000, 8.000000119e-001 );
00416     a[26] = lapack_make_complex_float( 0.000000000e+000, 0.000000000e+000 );
00417     a[3] = lapack_make_complex_float( -1.889999986e+000, 1.149999976e+000 );
00418     a[11] = lapack_make_complex_float( 3.999999911e-002, -3.690000057e+000 );
00419     a[19] = lapack_make_complex_float( -1.999999955e-002, 4.600000083e-001 );
00420     a[27] = lapack_make_complex_float( 3.300000131e-001, -2.599999905e-001 );
00421 }
00422 static void init_b( lapack_int size, lapack_complex_float *b ) {
00423     lapack_int i;
00424     for( i = 0; i < size; i++ ) {
00425         b[i] = lapack_make_complex_float( 0.0f, 0.0f );
00426     }
00427     b[0] = lapack_make_complex_float( -1.477999973e+001, -3.236000061e+001 );
00428     b[8] = lapack_make_complex_float( -1.802000046e+001, 2.845999908e+001 );
00429     b[1] = lapack_make_complex_float( 2.980000019e+000, -2.140000105e+000 );
00430     b[9] = lapack_make_complex_float( 1.422000027e+001, 1.542000008e+001 );
00431     b[2] = lapack_make_complex_float( -2.095999908e+001, 1.705999947e+001 );
00432     b[10] = lapack_make_complex_float( 5.619999886e+000, 3.588999939e+001 );
00433     b[3] = lapack_make_complex_float( 9.539999962e+000, 9.909999847e+000 );
00434     b[11] = lapack_make_complex_float( -1.645999908e+001, -1.730000019e+000 );
00435 }
00436 static void init_x( lapack_int size, lapack_complex_float *x ) {
00437     lapack_int i;
00438     for( i = 0; i < size; i++ ) {
00439         x[i] = lapack_make_complex_float( 0.0f, 0.0f );
00440     }
00441     x[0] = lapack_make_complex_float( -5.000000000e+000, -2.000000238e+000 );
00442     x[8] = lapack_make_complex_float( 9.999995232e-001, 5.000000000e+000 );
00443     x[1] = lapack_make_complex_float( -3.000000000e+000, -1.000000000e+000 );
00444     x[9] = lapack_make_complex_float( -2.000000238e+000, -2.000000000e+000 );
00445     x[2] = lapack_make_complex_float( 2.000000477e+000, 9.999999404e-001 );
00446     x[10] = lapack_make_complex_float( 3.000000000e+000, 3.999999046e+000 );
00447     x[3] = lapack_make_complex_float( 4.000001431e+000, 2.999997616e+000 );
00448     x[11] = lapack_make_complex_float( 4.000000000e+000, -3.000003576e+000 );
00449 }
00450 static void init_ferr( lapack_int size, float *ferr ) {
00451     lapack_int i;
00452     for( i = 0; i < size; i++ ) {
00453         ferr[i] = 0;
00454     }
00455 }
00456 static void init_berr( lapack_int size, float *berr ) {
00457     lapack_int i;
00458     for( i = 0; i < size; i++ ) {
00459         berr[i] = 0;
00460     }
00461 }
00462 static void init_work( lapack_int size, lapack_complex_float *work ) {
00463     lapack_int i;
00464     for( i = 0; i < size; i++ ) {
00465         work[i] = lapack_make_complex_float( 0.0f, 0.0f );
00466     }
00467 }
00468 static void init_rwork( lapack_int size, float *rwork ) {
00469     lapack_int i;
00470     for( i = 0; i < size; i++ ) {
00471         rwork[i] = 0;
00472     }
00473 }
00474 
00475 /* Auxiliary function: C interface to ctrrfs results check */
00476 /* Return value: 0 - test is passed, non-zero - test is failed */
00477 static int compare_ctrrfs( float *ferr, float *ferr_i, float *berr,
00478                            float *berr_i, lapack_int info, lapack_int info_i,
00479                            lapack_int nrhs )
00480 {
00481     lapack_int i;
00482     int failed = 0;
00483     for( i = 0; i < nrhs; i++ ) {
00484         failed += compare_floats(ferr[i],ferr_i[i]);
00485     }
00486     for( i = 0; i < nrhs; i++ ) {
00487         failed += compare_floats(berr[i],berr_i[i]);
00488     }
00489     failed += (info == info_i) ? 0 : 1;
00490     if( info != 0 || info_i != 0 ) {
00491         printf( "info=%d, info_i=%d\n",(int)info,(int)info_i );
00492     }
00493 
00494     return failed;
00495 }


swiftnav
Author(s):
autogenerated on Sat Jun 8 2019 18:55:35