ztprfs_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 * ztprfs_1 is the test program for the C interface to LAPACK
00036 * routine ztprfs
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_ztprfs( char *uplo, char *trans, char *diag,
00055                                  lapack_int *n, lapack_int *nrhs,
00056                                  lapack_int *ldb, lapack_int *ldx );
00057 static void init_ap( lapack_int size, lapack_complex_double *ap );
00058 static void init_b( lapack_int size, lapack_complex_double *b );
00059 static void init_x( lapack_int size, lapack_complex_double *x );
00060 static void init_ferr( lapack_int size, double *ferr );
00061 static void init_berr( lapack_int size, double *berr );
00062 static void init_work( lapack_int size, lapack_complex_double *work );
00063 static void init_rwork( lapack_int size, double *rwork );
00064 static int compare_ztprfs( double *ferr, double *ferr_i, double *berr,
00065                            double *berr_i, lapack_int info, lapack_int info_i,
00066                            lapack_int nrhs );
00067 
00068 int main(void)
00069 {
00070     /* Local scalars */
00071     char uplo, uplo_i;
00072     char trans, trans_i;
00073     char diag, diag_i;
00074     lapack_int n, n_i;
00075     lapack_int nrhs, nrhs_i;
00076     lapack_int ldb, ldb_i;
00077     lapack_int ldb_r;
00078     lapack_int ldx, ldx_i;
00079     lapack_int ldx_r;
00080     lapack_int info, info_i;
00081     lapack_int i;
00082     int failed;
00083 
00084     /* Local arrays */
00085     lapack_complex_double *ap = NULL, *ap_i = NULL;
00086     lapack_complex_double *b = NULL, *b_i = NULL;
00087     lapack_complex_double *x = NULL, *x_i = NULL;
00088     double *ferr = NULL, *ferr_i = NULL;
00089     double *berr = NULL, *berr_i = NULL;
00090     lapack_complex_double *work = NULL, *work_i = NULL;
00091     double *rwork = NULL, *rwork_i = NULL;
00092     double *ferr_save = NULL;
00093     double *berr_save = NULL;
00094     lapack_complex_double *ap_r = NULL;
00095     lapack_complex_double *b_r = NULL;
00096     lapack_complex_double *x_r = NULL;
00097 
00098     /* Iniitialize the scalar parameters */
00099     init_scalars_ztprfs( &uplo, &trans, &diag, &n, &nrhs, &ldb, &ldx );
00100     ldb_r = nrhs+2;
00101     ldx_r = nrhs+2;
00102     uplo_i = uplo;
00103     trans_i = trans;
00104     diag_i = diag;
00105     n_i = n;
00106     nrhs_i = nrhs;
00107     ldb_i = ldb;
00108     ldx_i = ldx;
00109 
00110     /* Allocate memory for the LAPACK routine arrays */
00111     ap = (lapack_complex_double *)
00112         LAPACKE_malloc( ((n*(n+1)/2)) * sizeof(lapack_complex_double) );
00113     b = (lapack_complex_double *)
00114         LAPACKE_malloc( ldb*nrhs * sizeof(lapack_complex_double) );
00115     x = (lapack_complex_double *)
00116         LAPACKE_malloc( ldx*nrhs * sizeof(lapack_complex_double) );
00117     ferr = (double *)LAPACKE_malloc( nrhs * sizeof(double) );
00118     berr = (double *)LAPACKE_malloc( nrhs * sizeof(double) );
00119     work = (lapack_complex_double *)
00120         LAPACKE_malloc( 2*n * sizeof(lapack_complex_double) );
00121     rwork = (double *)LAPACKE_malloc( n * sizeof(double) );
00122 
00123     /* Allocate memory for the C interface function arrays */
00124     ap_i = (lapack_complex_double *)
00125         LAPACKE_malloc( ((n*(n+1)/2)) * sizeof(lapack_complex_double) );
00126     b_i = (lapack_complex_double *)
00127         LAPACKE_malloc( ldb*nrhs * sizeof(lapack_complex_double) );
00128     x_i = (lapack_complex_double *)
00129         LAPACKE_malloc( ldx*nrhs * sizeof(lapack_complex_double) );
00130     ferr_i = (double *)LAPACKE_malloc( nrhs * sizeof(double) );
00131     berr_i = (double *)LAPACKE_malloc( nrhs * sizeof(double) );
00132     work_i = (lapack_complex_double *)
00133         LAPACKE_malloc( 2*n * sizeof(lapack_complex_double) );
00134     rwork_i = (double *)LAPACKE_malloc( n * sizeof(double) );
00135 
00136     /* Allocate memory for the backup arrays */
00137     ferr_save = (double *)LAPACKE_malloc( nrhs * sizeof(double) );
00138     berr_save = (double *)LAPACKE_malloc( nrhs * sizeof(double) );
00139 
00140     /* Allocate memory for the row-major arrays */
00141     ap_r = (lapack_complex_double *)
00142         LAPACKE_malloc( n*(n+1)/2 * sizeof(lapack_complex_double) );
00143     b_r = (lapack_complex_double *)
00144         LAPACKE_malloc( n*(nrhs+2) * sizeof(lapack_complex_double) );
00145     x_r = (lapack_complex_double *)
00146         LAPACKE_malloc( n*(nrhs+2) * sizeof(lapack_complex_double) );
00147 
00148     /* Initialize input arrays */
00149     init_ap( (n*(n+1)/2), ap );
00150     init_b( ldb*nrhs, b );
00151     init_x( ldx*nrhs, x );
00152     init_ferr( nrhs, ferr );
00153     init_berr( nrhs, berr );
00154     init_work( 2*n, work );
00155     init_rwork( n, rwork );
00156 
00157     /* Backup the ouptut arrays */
00158     for( i = 0; i < nrhs; i++ ) {
00159         ferr_save[i] = ferr[i];
00160     }
00161     for( i = 0; i < nrhs; i++ ) {
00162         berr_save[i] = berr[i];
00163     }
00164 
00165     /* Call the LAPACK routine */
00166     ztprfs_( &uplo, &trans, &diag, &n, &nrhs, ap, b, &ldb, x, &ldx, ferr, berr,
00167              work, rwork, &info );
00168 
00169     /* Initialize input data, call the column-major middle-level
00170      * interface to LAPACK routine and check the results */
00171     for( i = 0; i < (n*(n+1)/2); i++ ) {
00172         ap_i[i] = ap[i];
00173     }
00174     for( i = 0; i < ldb*nrhs; i++ ) {
00175         b_i[i] = b[i];
00176     }
00177     for( i = 0; i < ldx*nrhs; i++ ) {
00178         x_i[i] = x[i];
00179     }
00180     for( i = 0; i < nrhs; i++ ) {
00181         ferr_i[i] = ferr_save[i];
00182     }
00183     for( i = 0; i < nrhs; i++ ) {
00184         berr_i[i] = berr_save[i];
00185     }
00186     for( i = 0; i < 2*n; i++ ) {
00187         work_i[i] = work[i];
00188     }
00189     for( i = 0; i < n; i++ ) {
00190         rwork_i[i] = rwork[i];
00191     }
00192     info_i = LAPACKE_ztprfs_work( LAPACK_COL_MAJOR, uplo_i, trans_i, diag_i,
00193                                   n_i, nrhs_i, ap_i, b_i, ldb_i, x_i, ldx_i,
00194                                   ferr_i, berr_i, work_i, rwork_i );
00195 
00196     failed = compare_ztprfs( ferr, ferr_i, berr, berr_i, info, info_i, nrhs );
00197     if( failed == 0 ) {
00198         printf( "PASSED: column-major middle-level interface to ztprfs\n" );
00199     } else {
00200         printf( "FAILED: column-major middle-level interface to ztprfs\n" );
00201     }
00202 
00203     /* Initialize input data, call the column-major high-level
00204      * interface to LAPACK routine and check the results */
00205     for( i = 0; i < (n*(n+1)/2); i++ ) {
00206         ap_i[i] = ap[i];
00207     }
00208     for( i = 0; i < ldb*nrhs; i++ ) {
00209         b_i[i] = b[i];
00210     }
00211     for( i = 0; i < ldx*nrhs; i++ ) {
00212         x_i[i] = x[i];
00213     }
00214     for( i = 0; i < nrhs; i++ ) {
00215         ferr_i[i] = ferr_save[i];
00216     }
00217     for( i = 0; i < nrhs; i++ ) {
00218         berr_i[i] = berr_save[i];
00219     }
00220     for( i = 0; i < 2*n; i++ ) {
00221         work_i[i] = work[i];
00222     }
00223     for( i = 0; i < n; i++ ) {
00224         rwork_i[i] = rwork[i];
00225     }
00226     info_i = LAPACKE_ztprfs( LAPACK_COL_MAJOR, uplo_i, trans_i, diag_i, n_i,
00227                              nrhs_i, ap_i, b_i, ldb_i, x_i, ldx_i, ferr_i,
00228                              berr_i );
00229 
00230     failed = compare_ztprfs( ferr, ferr_i, berr, berr_i, info, info_i, nrhs );
00231     if( failed == 0 ) {
00232         printf( "PASSED: column-major high-level interface to ztprfs\n" );
00233     } else {
00234         printf( "FAILED: column-major high-level interface to ztprfs\n" );
00235     }
00236 
00237     /* Initialize input data, call the row-major middle-level
00238      * interface to LAPACK routine and check the results */
00239     for( i = 0; i < (n*(n+1)/2); i++ ) {
00240         ap_i[i] = ap[i];
00241     }
00242     for( i = 0; i < ldb*nrhs; i++ ) {
00243         b_i[i] = b[i];
00244     }
00245     for( i = 0; i < ldx*nrhs; i++ ) {
00246         x_i[i] = x[i];
00247     }
00248     for( i = 0; i < nrhs; i++ ) {
00249         ferr_i[i] = ferr_save[i];
00250     }
00251     for( i = 0; i < nrhs; i++ ) {
00252         berr_i[i] = berr_save[i];
00253     }
00254     for( i = 0; i < 2*n; i++ ) {
00255         work_i[i] = work[i];
00256     }
00257     for( i = 0; i < n; i++ ) {
00258         rwork_i[i] = rwork[i];
00259     }
00260 
00261     LAPACKE_zpp_trans( LAPACK_COL_MAJOR, uplo, n, ap_i, ap_r );
00262     LAPACKE_zge_trans( LAPACK_COL_MAJOR, n, nrhs, b_i, ldb, b_r, nrhs+2 );
00263     LAPACKE_zge_trans( LAPACK_COL_MAJOR, n, nrhs, x_i, ldx, x_r, nrhs+2 );
00264     info_i = LAPACKE_ztprfs_work( LAPACK_ROW_MAJOR, uplo_i, trans_i, diag_i,
00265                                   n_i, nrhs_i, ap_r, b_r, ldb_r, x_r, ldx_r,
00266                                   ferr_i, berr_i, work_i, rwork_i );
00267 
00268     failed = compare_ztprfs( ferr, ferr_i, berr, berr_i, info, info_i, nrhs );
00269     if( failed == 0 ) {
00270         printf( "PASSED: row-major middle-level interface to ztprfs\n" );
00271     } else {
00272         printf( "FAILED: row-major middle-level interface to ztprfs\n" );
00273     }
00274 
00275     /* Initialize input data, call the row-major high-level
00276      * interface to LAPACK routine and check the results */
00277     for( i = 0; i < (n*(n+1)/2); i++ ) {
00278         ap_i[i] = ap[i];
00279     }
00280     for( i = 0; i < ldb*nrhs; i++ ) {
00281         b_i[i] = b[i];
00282     }
00283     for( i = 0; i < ldx*nrhs; i++ ) {
00284         x_i[i] = x[i];
00285     }
00286     for( i = 0; i < nrhs; i++ ) {
00287         ferr_i[i] = ferr_save[i];
00288     }
00289     for( i = 0; i < nrhs; i++ ) {
00290         berr_i[i] = berr_save[i];
00291     }
00292     for( i = 0; i < 2*n; i++ ) {
00293         work_i[i] = work[i];
00294     }
00295     for( i = 0; i < n; i++ ) {
00296         rwork_i[i] = rwork[i];
00297     }
00298 
00299     /* Init row_major arrays */
00300     LAPACKE_zpp_trans( LAPACK_COL_MAJOR, uplo, n, ap_i, ap_r );
00301     LAPACKE_zge_trans( LAPACK_COL_MAJOR, n, nrhs, b_i, ldb, b_r, nrhs+2 );
00302     LAPACKE_zge_trans( LAPACK_COL_MAJOR, n, nrhs, x_i, ldx, x_r, nrhs+2 );
00303     info_i = LAPACKE_ztprfs( LAPACK_ROW_MAJOR, uplo_i, trans_i, diag_i, n_i,
00304                              nrhs_i, ap_r, b_r, ldb_r, x_r, ldx_r, ferr_i,
00305                              berr_i );
00306 
00307     failed = compare_ztprfs( ferr, ferr_i, berr, berr_i, info, info_i, nrhs );
00308     if( failed == 0 ) {
00309         printf( "PASSED: row-major high-level interface to ztprfs\n" );
00310     } else {
00311         printf( "FAILED: row-major high-level interface to ztprfs\n" );
00312     }
00313 
00314     /* Release memory */
00315     if( ap != NULL ) {
00316         LAPACKE_free( ap );
00317     }
00318     if( ap_i != NULL ) {
00319         LAPACKE_free( ap_i );
00320     }
00321     if( ap_r != NULL ) {
00322         LAPACKE_free( ap_r );
00323     }
00324     if( b != NULL ) {
00325         LAPACKE_free( b );
00326     }
00327     if( b_i != NULL ) {
00328         LAPACKE_free( b_i );
00329     }
00330     if( b_r != NULL ) {
00331         LAPACKE_free( b_r );
00332     }
00333     if( x != NULL ) {
00334         LAPACKE_free( x );
00335     }
00336     if( x_i != NULL ) {
00337         LAPACKE_free( x_i );
00338     }
00339     if( x_r != NULL ) {
00340         LAPACKE_free( x_r );
00341     }
00342     if( ferr != NULL ) {
00343         LAPACKE_free( ferr );
00344     }
00345     if( ferr_i != NULL ) {
00346         LAPACKE_free( ferr_i );
00347     }
00348     if( ferr_save != NULL ) {
00349         LAPACKE_free( ferr_save );
00350     }
00351     if( berr != NULL ) {
00352         LAPACKE_free( berr );
00353     }
00354     if( berr_i != NULL ) {
00355         LAPACKE_free( berr_i );
00356     }
00357     if( berr_save != NULL ) {
00358         LAPACKE_free( berr_save );
00359     }
00360     if( work != NULL ) {
00361         LAPACKE_free( work );
00362     }
00363     if( work_i != NULL ) {
00364         LAPACKE_free( work_i );
00365     }
00366     if( rwork != NULL ) {
00367         LAPACKE_free( rwork );
00368     }
00369     if( rwork_i != NULL ) {
00370         LAPACKE_free( rwork_i );
00371     }
00372 
00373     return 0;
00374 }
00375 
00376 /* Auxiliary function: ztprfs scalar parameters initialization */
00377 static void init_scalars_ztprfs( char *uplo, char *trans, char *diag,
00378                                  lapack_int *n, lapack_int *nrhs,
00379                                  lapack_int *ldb, lapack_int *ldx )
00380 {
00381     *uplo = 'L';
00382     *trans = 'N';
00383     *diag = 'N';
00384     *n = 4;
00385     *nrhs = 2;
00386     *ldb = 8;
00387     *ldx = 8;
00388 
00389     return;
00390 }
00391 
00392 /* Auxiliary functions: ztprfs array parameters initialization */
00393 static void init_ap( lapack_int size, lapack_complex_double *ap ) {
00394     lapack_int i;
00395     for( i = 0; i < size; i++ ) {
00396         ap[i] = lapack_make_complex_double( 0.0, 0.0 );
00397     }
00398     ap[0] = lapack_make_complex_double( 4.78000000000000020e+000,
00399                                         4.55999999999999960e+000 );
00400     ap[1] = lapack_make_complex_double( 2.00000000000000000e+000,
00401                                         -2.99999999999999990e-001 );
00402     ap[2] = lapack_make_complex_double( 2.89000000000000010e+000,
00403                                         -1.34000000000000010e+000 );
00404     ap[3] = lapack_make_complex_double( -1.88999999999999990e+000,
00405                                         1.14999999999999990e+000 );
00406     ap[4] = lapack_make_complex_double( -4.11000000000000030e+000,
00407                                         1.25000000000000000e+000 );
00408     ap[5] = lapack_make_complex_double( 2.35999999999999990e+000,
00409                                         -4.25000000000000000e+000 );
00410     ap[6] = lapack_make_complex_double( 4.00000000000000010e-002,
00411                                         -3.68999999999999990e+000 );
00412     ap[7] = lapack_make_complex_double( 4.15000000000000040e+000,
00413                                         8.00000000000000040e-001 );
00414     ap[8] = lapack_make_complex_double( -2.00000000000000000e-002,
00415                                         4.60000000000000020e-001 );
00416     ap[9] = lapack_make_complex_double( 3.30000000000000020e-001,
00417                                         -2.60000000000000010e-001 );
00418 }
00419 static void init_b( lapack_int size, lapack_complex_double *b ) {
00420     lapack_int i;
00421     for( i = 0; i < size; i++ ) {
00422         b[i] = lapack_make_complex_double( 0.0, 0.0 );
00423     }
00424     b[0] = lapack_make_complex_double( -1.47799999999999990e+001,
00425                                        -3.23599999999999990e+001 );
00426     b[8] = lapack_make_complex_double( -1.80200000000000000e+001,
00427                                        2.84600000000000010e+001 );
00428     b[1] = lapack_make_complex_double( 2.98000000000000000e+000,
00429                                        -2.14000000000000010e+000 );
00430     b[9] = lapack_make_complex_double( 1.42200000000000010e+001,
00431                                        1.54200000000000000e+001 );
00432     b[2] = lapack_make_complex_double( -2.09600000000000010e+001,
00433                                        1.70599999999999990e+001 );
00434     b[10] = lapack_make_complex_double( 5.62000000000000010e+000,
00435                                         3.58900000000000010e+001 );
00436     b[3] = lapack_make_complex_double( 9.53999999999999910e+000,
00437                                        9.91000000000000010e+000 );
00438     b[11] = lapack_make_complex_double( -1.64600000000000010e+001,
00439                                         -1.73000000000000000e+000 );
00440 }
00441 static void init_x( lapack_int size, lapack_complex_double *x ) {
00442     lapack_int i;
00443     for( i = 0; i < size; i++ ) {
00444         x[i] = lapack_make_complex_double( 0.0, 0.0 );
00445     }
00446     x[0] = lapack_make_complex_double( -5.00000000000000000e+000,
00447                                        -2.00000000000000090e+000 );
00448     x[8] = lapack_make_complex_double( 1.00000000000000040e+000,
00449                                        5.00000000000000000e+000 );
00450     x[1] = lapack_make_complex_double( -2.99999999999999960e+000,
00451                                        -1.00000000000000040e+000 );
00452     x[9] = lapack_make_complex_double( -1.99999999999999960e+000,
00453                                        -1.99999999999999960e+000 );
00454     x[2] = lapack_make_complex_double( 2.00000000000000040e+000,
00455                                        1.00000000000000090e+000 );
00456     x[10] = lapack_make_complex_double( 2.99999999999999910e+000,
00457                                         4.00000000000000000e+000 );
00458     x[3] = lapack_make_complex_double( 4.00000000000000360e+000,
00459                                        3.00000000000000090e+000 );
00460     x[11] = lapack_make_complex_double( 3.99999999999999510e+000,
00461                                         -3.00000000000000130e+000 );
00462 }
00463 static void init_ferr( lapack_int size, double *ferr ) {
00464     lapack_int i;
00465     for( i = 0; i < size; i++ ) {
00466         ferr[i] = 0;
00467     }
00468 }
00469 static void init_berr( lapack_int size, double *berr ) {
00470     lapack_int i;
00471     for( i = 0; i < size; i++ ) {
00472         berr[i] = 0;
00473     }
00474 }
00475 static void init_work( lapack_int size, lapack_complex_double *work ) {
00476     lapack_int i;
00477     for( i = 0; i < size; i++ ) {
00478         work[i] = lapack_make_complex_double( 0.0, 0.0 );
00479     }
00480 }
00481 static void init_rwork( lapack_int size, double *rwork ) {
00482     lapack_int i;
00483     for( i = 0; i < size; i++ ) {
00484         rwork[i] = 0;
00485     }
00486 }
00487 
00488 /* Auxiliary function: C interface to ztprfs results check */
00489 /* Return value: 0 - test is passed, non-zero - test is failed */
00490 static int compare_ztprfs( double *ferr, double *ferr_i, double *berr,
00491                            double *berr_i, lapack_int info, lapack_int info_i,
00492                            lapack_int nrhs )
00493 {
00494     lapack_int i;
00495     int failed = 0;
00496     for( i = 0; i < nrhs; i++ ) {
00497         failed += compare_doubles(ferr[i],ferr_i[i]);
00498     }
00499     for( i = 0; i < nrhs; i++ ) {
00500         failed += compare_doubles(berr[i],berr_i[i]);
00501     }
00502     failed += (info == info_i) ? 0 : 1;
00503     if( info != 0 || info_i != 0 ) {
00504         printf( "info=%d, info_i=%d\n",(int)info,(int)info_i );
00505     }
00506 
00507     return failed;
00508 }


swiftnav
Author(s):
autogenerated on Sat Jun 8 2019 18:56:44