ztrtrs_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 * ztrtrs_1 is the test program for the C interface to LAPACK
00036 * routine ztrtrs
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_ztrtrs( char *uplo, char *trans, char *diag,
00055                                  lapack_int *n, lapack_int *nrhs,
00056                                  lapack_int *lda, lapack_int *ldb );
00057 static void init_a( lapack_int size, lapack_complex_double *a );
00058 static void init_b( lapack_int size, lapack_complex_double *b );
00059 static int compare_ztrtrs( lapack_complex_double *b, lapack_complex_double *b_i,
00060                            lapack_int info, lapack_int info_i, lapack_int ldb,
00061                            lapack_int nrhs );
00062 
00063 int main(void)
00064 {
00065     /* Local scalars */
00066     char uplo, uplo_i;
00067     char trans, trans_i;
00068     char diag, diag_i;
00069     lapack_int n, n_i;
00070     lapack_int nrhs, nrhs_i;
00071     lapack_int lda, lda_i;
00072     lapack_int lda_r;
00073     lapack_int ldb, ldb_i;
00074     lapack_int ldb_r;
00075     lapack_int info, info_i;
00076     lapack_int i;
00077     int failed;
00078 
00079     /* Local arrays */
00080     lapack_complex_double *a = NULL, *a_i = NULL;
00081     lapack_complex_double *b = NULL, *b_i = NULL;
00082     lapack_complex_double *b_save = NULL;
00083     lapack_complex_double *a_r = NULL;
00084     lapack_complex_double *b_r = NULL;
00085 
00086     /* Iniitialize the scalar parameters */
00087     init_scalars_ztrtrs( &uplo, &trans, &diag, &n, &nrhs, &lda, &ldb );
00088     lda_r = n+2;
00089     ldb_r = nrhs+2;
00090     uplo_i = uplo;
00091     trans_i = trans;
00092     diag_i = diag;
00093     n_i = n;
00094     nrhs_i = nrhs;
00095     lda_i = lda;
00096     ldb_i = ldb;
00097 
00098     /* Allocate memory for the LAPACK routine arrays */
00099     a = (lapack_complex_double *)
00100         LAPACKE_malloc( lda*n * sizeof(lapack_complex_double) );
00101     b = (lapack_complex_double *)
00102         LAPACKE_malloc( ldb*nrhs * sizeof(lapack_complex_double) );
00103 
00104     /* Allocate memory for the C interface function arrays */
00105     a_i = (lapack_complex_double *)
00106         LAPACKE_malloc( lda*n * sizeof(lapack_complex_double) );
00107     b_i = (lapack_complex_double *)
00108         LAPACKE_malloc( ldb*nrhs * sizeof(lapack_complex_double) );
00109 
00110     /* Allocate memory for the backup arrays */
00111     b_save = (lapack_complex_double *)
00112         LAPACKE_malloc( ldb*nrhs * sizeof(lapack_complex_double) );
00113 
00114     /* Allocate memory for the row-major arrays */
00115     a_r = (lapack_complex_double *)
00116         LAPACKE_malloc( n*(n+2) * sizeof(lapack_complex_double) );
00117     b_r = (lapack_complex_double *)
00118         LAPACKE_malloc( n*(nrhs+2) * sizeof(lapack_complex_double) );
00119 
00120     /* Initialize input arrays */
00121     init_a( lda*n, a );
00122     init_b( ldb*nrhs, b );
00123 
00124     /* Backup the ouptut arrays */
00125     for( i = 0; i < ldb*nrhs; i++ ) {
00126         b_save[i] = b[i];
00127     }
00128 
00129     /* Call the LAPACK routine */
00130     ztrtrs_( &uplo, &trans, &diag, &n, &nrhs, a, &lda, b, &ldb, &info );
00131 
00132     /* Initialize input data, call the column-major middle-level
00133      * interface to LAPACK routine and check the results */
00134     for( i = 0; i < lda*n; i++ ) {
00135         a_i[i] = a[i];
00136     }
00137     for( i = 0; i < ldb*nrhs; i++ ) {
00138         b_i[i] = b_save[i];
00139     }
00140     info_i = LAPACKE_ztrtrs_work( LAPACK_COL_MAJOR, uplo_i, trans_i, diag_i,
00141                                   n_i, nrhs_i, a_i, lda_i, b_i, ldb_i );
00142 
00143     failed = compare_ztrtrs( b, b_i, info, info_i, ldb, nrhs );
00144     if( failed == 0 ) {
00145         printf( "PASSED: column-major middle-level interface to ztrtrs\n" );
00146     } else {
00147         printf( "FAILED: column-major middle-level interface to ztrtrs\n" );
00148     }
00149 
00150     /* Initialize input data, call the column-major high-level
00151      * interface to LAPACK routine and check the results */
00152     for( i = 0; i < lda*n; i++ ) {
00153         a_i[i] = a[i];
00154     }
00155     for( i = 0; i < ldb*nrhs; i++ ) {
00156         b_i[i] = b_save[i];
00157     }
00158     info_i = LAPACKE_ztrtrs( LAPACK_COL_MAJOR, uplo_i, trans_i, diag_i, n_i,
00159                              nrhs_i, a_i, lda_i, b_i, ldb_i );
00160 
00161     failed = compare_ztrtrs( b, b_i, info, info_i, ldb, nrhs );
00162     if( failed == 0 ) {
00163         printf( "PASSED: column-major high-level interface to ztrtrs\n" );
00164     } else {
00165         printf( "FAILED: column-major high-level interface to ztrtrs\n" );
00166     }
00167 
00168     /* Initialize input data, call the row-major middle-level
00169      * interface to LAPACK routine and check the results */
00170     for( i = 0; i < lda*n; i++ ) {
00171         a_i[i] = a[i];
00172     }
00173     for( i = 0; i < ldb*nrhs; i++ ) {
00174         b_i[i] = b_save[i];
00175     }
00176 
00177     LAPACKE_zge_trans( LAPACK_COL_MAJOR, n, n, a_i, lda, a_r, n+2 );
00178     LAPACKE_zge_trans( LAPACK_COL_MAJOR, n, nrhs, b_i, ldb, b_r, nrhs+2 );
00179     info_i = LAPACKE_ztrtrs_work( LAPACK_ROW_MAJOR, uplo_i, trans_i, diag_i,
00180                                   n_i, nrhs_i, a_r, lda_r, b_r, ldb_r );
00181 
00182     LAPACKE_zge_trans( LAPACK_ROW_MAJOR, n, nrhs, b_r, nrhs+2, b_i, ldb );
00183 
00184     failed = compare_ztrtrs( b, b_i, info, info_i, ldb, nrhs );
00185     if( failed == 0 ) {
00186         printf( "PASSED: row-major middle-level interface to ztrtrs\n" );
00187     } else {
00188         printf( "FAILED: row-major middle-level interface to ztrtrs\n" );
00189     }
00190 
00191     /* Initialize input data, call the row-major high-level
00192      * interface to LAPACK routine and check the results */
00193     for( i = 0; i < lda*n; i++ ) {
00194         a_i[i] = a[i];
00195     }
00196     for( i = 0; i < ldb*nrhs; i++ ) {
00197         b_i[i] = b_save[i];
00198     }
00199 
00200     /* Init row_major arrays */
00201     LAPACKE_zge_trans( LAPACK_COL_MAJOR, n, n, a_i, lda, a_r, n+2 );
00202     LAPACKE_zge_trans( LAPACK_COL_MAJOR, n, nrhs, b_i, ldb, b_r, nrhs+2 );
00203     info_i = LAPACKE_ztrtrs( LAPACK_ROW_MAJOR, uplo_i, trans_i, diag_i, n_i,
00204                              nrhs_i, a_r, lda_r, b_r, ldb_r );
00205 
00206     LAPACKE_zge_trans( LAPACK_ROW_MAJOR, n, nrhs, b_r, nrhs+2, b_i, ldb );
00207 
00208     failed = compare_ztrtrs( b, b_i, info, info_i, ldb, nrhs );
00209     if( failed == 0 ) {
00210         printf( "PASSED: row-major high-level interface to ztrtrs\n" );
00211     } else {
00212         printf( "FAILED: row-major high-level interface to ztrtrs\n" );
00213     }
00214 
00215     /* Release memory */
00216     if( a != NULL ) {
00217         LAPACKE_free( a );
00218     }
00219     if( a_i != NULL ) {
00220         LAPACKE_free( a_i );
00221     }
00222     if( a_r != NULL ) {
00223         LAPACKE_free( a_r );
00224     }
00225     if( b != NULL ) {
00226         LAPACKE_free( b );
00227     }
00228     if( b_i != NULL ) {
00229         LAPACKE_free( b_i );
00230     }
00231     if( b_r != NULL ) {
00232         LAPACKE_free( b_r );
00233     }
00234     if( b_save != NULL ) {
00235         LAPACKE_free( b_save );
00236     }
00237 
00238     return 0;
00239 }
00240 
00241 /* Auxiliary function: ztrtrs scalar parameters initialization */
00242 static void init_scalars_ztrtrs( char *uplo, char *trans, char *diag,
00243                                  lapack_int *n, lapack_int *nrhs,
00244                                  lapack_int *lda, lapack_int *ldb )
00245 {
00246     *uplo = 'L';
00247     *trans = 'N';
00248     *diag = 'N';
00249     *n = 4;
00250     *nrhs = 2;
00251     *lda = 8;
00252     *ldb = 8;
00253 
00254     return;
00255 }
00256 
00257 /* Auxiliary functions: ztrtrs array parameters initialization */
00258 static void init_a( lapack_int size, lapack_complex_double *a ) {
00259     lapack_int i;
00260     for( i = 0; i < size; i++ ) {
00261         a[i] = lapack_make_complex_double( 0.0, 0.0 );
00262     }
00263     a[0] = lapack_make_complex_double( 4.78000000000000020e+000,
00264                                        4.55999999999999960e+000 );
00265     a[8] = lapack_make_complex_double( 0.00000000000000000e+000,
00266                                        0.00000000000000000e+000 );
00267     a[16] = lapack_make_complex_double( 0.00000000000000000e+000,
00268                                         0.00000000000000000e+000 );
00269     a[24] = lapack_make_complex_double( 0.00000000000000000e+000,
00270                                         0.00000000000000000e+000 );
00271     a[1] = lapack_make_complex_double( 2.00000000000000000e+000,
00272                                        -2.99999999999999990e-001 );
00273     a[9] = lapack_make_complex_double( -4.11000000000000030e+000,
00274                                        1.25000000000000000e+000 );
00275     a[17] = lapack_make_complex_double( 0.00000000000000000e+000,
00276                                         0.00000000000000000e+000 );
00277     a[25] = lapack_make_complex_double( 0.00000000000000000e+000,
00278                                         0.00000000000000000e+000 );
00279     a[2] = lapack_make_complex_double( 2.89000000000000010e+000,
00280                                        -1.34000000000000010e+000 );
00281     a[10] = lapack_make_complex_double( 2.35999999999999990e+000,
00282                                         -4.25000000000000000e+000 );
00283     a[18] = lapack_make_complex_double( 4.15000000000000040e+000,
00284                                         8.00000000000000040e-001 );
00285     a[26] = lapack_make_complex_double( 0.00000000000000000e+000,
00286                                         0.00000000000000000e+000 );
00287     a[3] = lapack_make_complex_double( -1.88999999999999990e+000,
00288                                        1.14999999999999990e+000 );
00289     a[11] = lapack_make_complex_double( 4.00000000000000010e-002,
00290                                         -3.68999999999999990e+000 );
00291     a[19] = lapack_make_complex_double( -2.00000000000000000e-002,
00292                                         4.60000000000000020e-001 );
00293     a[27] = lapack_make_complex_double( 3.30000000000000020e-001,
00294                                         -2.60000000000000010e-001 );
00295 }
00296 static void init_b( lapack_int size, lapack_complex_double *b ) {
00297     lapack_int i;
00298     for( i = 0; i < size; i++ ) {
00299         b[i] = lapack_make_complex_double( 0.0, 0.0 );
00300     }
00301     b[0] = lapack_make_complex_double( -1.47799999999999990e+001,
00302                                        -3.23599999999999990e+001 );
00303     b[8] = lapack_make_complex_double( -1.80200000000000000e+001,
00304                                        2.84600000000000010e+001 );
00305     b[1] = lapack_make_complex_double( 2.98000000000000000e+000,
00306                                        -2.14000000000000010e+000 );
00307     b[9] = lapack_make_complex_double( 1.42200000000000010e+001,
00308                                        1.54200000000000000e+001 );
00309     b[2] = lapack_make_complex_double( -2.09600000000000010e+001,
00310                                        1.70599999999999990e+001 );
00311     b[10] = lapack_make_complex_double( 5.62000000000000010e+000,
00312                                         3.58900000000000010e+001 );
00313     b[3] = lapack_make_complex_double( 9.53999999999999910e+000,
00314                                        9.91000000000000010e+000 );
00315     b[11] = lapack_make_complex_double( -1.64600000000000010e+001,
00316                                         -1.73000000000000000e+000 );
00317 }
00318 
00319 /* Auxiliary function: C interface to ztrtrs results check */
00320 /* Return value: 0 - test is passed, non-zero - test is failed */
00321 static int compare_ztrtrs( lapack_complex_double *b, lapack_complex_double *b_i,
00322                            lapack_int info, lapack_int info_i, lapack_int ldb,
00323                            lapack_int nrhs )
00324 {
00325     lapack_int i;
00326     int failed = 0;
00327     for( i = 0; i < ldb*nrhs; i++ ) {
00328         failed += compare_complex_doubles(b[i],b_i[i]);
00329     }
00330     failed += (info == info_i) ? 0 : 1;
00331     if( info != 0 || info_i != 0 ) {
00332         printf( "info=%d, info_i=%d\n",(int)info,(int)info_i );
00333     }
00334 
00335     return failed;
00336 }


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