csteqr_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 * csteqr_1 is the test program for the C interface to LAPACK
00036 * routine csteqr
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_csteqr( char *compz, lapack_int *n, lapack_int *ldz );
00055 static void init_d( lapack_int size, float *d );
00056 static void init_e( lapack_int size, float *e );
00057 static void init_z( lapack_int size, lapack_complex_float *z );
00058 static void init_work( lapack_int size, float *work );
00059 static int compare_csteqr( float *d, float *d_i, float *e, float *e_i,
00060                            lapack_complex_float *z, lapack_complex_float *z_i,
00061                            lapack_int info, lapack_int info_i, char compz,
00062                            lapack_int ldz, lapack_int n );
00063 
00064 int main(void)
00065 {
00066     /* Local scalars */
00067     char compz, compz_i;
00068     lapack_int n, n_i;
00069     lapack_int ldz, ldz_i;
00070     lapack_int ldz_r;
00071     lapack_int info, info_i;
00072     lapack_int i;
00073     int failed;
00074 
00075     /* Local arrays */
00076     float *d = NULL, *d_i = NULL;
00077     float *e = NULL, *e_i = NULL;
00078     lapack_complex_float *z = NULL, *z_i = NULL;
00079     float *work = NULL, *work_i = NULL;
00080     float *d_save = NULL;
00081     float *e_save = NULL;
00082     lapack_complex_float *z_save = NULL;
00083     lapack_complex_float *z_r = NULL;
00084 
00085     /* Iniitialize the scalar parameters */
00086     init_scalars_csteqr( &compz, &n, &ldz );
00087     ldz_r = n+2;
00088     compz_i = compz;
00089     n_i = n;
00090     ldz_i = ldz;
00091 
00092     /* Allocate memory for the LAPACK routine arrays */
00093     d = (float *)LAPACKE_malloc( n * sizeof(float) );
00094     e = (float *)LAPACKE_malloc( (n-1) * sizeof(float) );
00095     z = (lapack_complex_float *)
00096         LAPACKE_malloc( ldz*n * sizeof(lapack_complex_float) );
00097     work = (float *)LAPACKE_malloc( ((MAX(1,2*n-2))) * sizeof(float) );
00098 
00099     /* Allocate memory for the C interface function arrays */
00100     d_i = (float *)LAPACKE_malloc( n * sizeof(float) );
00101     e_i = (float *)LAPACKE_malloc( (n-1) * sizeof(float) );
00102     z_i = (lapack_complex_float *)
00103         LAPACKE_malloc( ldz*n * sizeof(lapack_complex_float) );
00104     work_i = (float *)LAPACKE_malloc( ((MAX(1,2*n-2))) * sizeof(float) );
00105 
00106     /* Allocate memory for the backup arrays */
00107     d_save = (float *)LAPACKE_malloc( n * sizeof(float) );
00108     e_save = (float *)LAPACKE_malloc( (n-1) * sizeof(float) );
00109     z_save = (lapack_complex_float *)
00110         LAPACKE_malloc( ldz*n * sizeof(lapack_complex_float) );
00111 
00112     /* Allocate memory for the row-major arrays */
00113     z_r = (lapack_complex_float *)
00114         LAPACKE_malloc( n*(n+2) * sizeof(lapack_complex_float) );
00115 
00116     /* Initialize input arrays */
00117     init_d( n, d );
00118     init_e( (n-1), e );
00119     init_z( ldz*n, z );
00120     init_work( (MAX(1,2*n-2)), work );
00121 
00122     /* Backup the ouptut arrays */
00123     for( i = 0; i < n; i++ ) {
00124         d_save[i] = d[i];
00125     }
00126     for( i = 0; i < (n-1); i++ ) {
00127         e_save[i] = e[i];
00128     }
00129     for( i = 0; i < ldz*n; i++ ) {
00130         z_save[i] = z[i];
00131     }
00132 
00133     /* Call the LAPACK routine */
00134     csteqr_( &compz, &n, d, e, z, &ldz, work, &info );
00135 
00136     /* Initialize input data, call the column-major middle-level
00137      * interface to LAPACK routine and check the results */
00138     for( i = 0; i < n; i++ ) {
00139         d_i[i] = d_save[i];
00140     }
00141     for( i = 0; i < (n-1); i++ ) {
00142         e_i[i] = e_save[i];
00143     }
00144     for( i = 0; i < ldz*n; i++ ) {
00145         z_i[i] = z_save[i];
00146     }
00147     for( i = 0; i < (MAX(1,2*n-2)); i++ ) {
00148         work_i[i] = work[i];
00149     }
00150     info_i = LAPACKE_csteqr_work( LAPACK_COL_MAJOR, compz_i, n_i, d_i, e_i, z_i,
00151                                   ldz_i, work_i );
00152 
00153     failed = compare_csteqr( d, d_i, e, e_i, z, z_i, info, info_i, compz, ldz,
00154                              n );
00155     if( failed == 0 ) {
00156         printf( "PASSED: column-major middle-level interface to csteqr\n" );
00157     } else {
00158         printf( "FAILED: column-major middle-level interface to csteqr\n" );
00159     }
00160 
00161     /* Initialize input data, call the column-major high-level
00162      * interface to LAPACK routine and check the results */
00163     for( i = 0; i < n; i++ ) {
00164         d_i[i] = d_save[i];
00165     }
00166     for( i = 0; i < (n-1); i++ ) {
00167         e_i[i] = e_save[i];
00168     }
00169     for( i = 0; i < ldz*n; i++ ) {
00170         z_i[i] = z_save[i];
00171     }
00172     for( i = 0; i < (MAX(1,2*n-2)); i++ ) {
00173         work_i[i] = work[i];
00174     }
00175     info_i = LAPACKE_csteqr( LAPACK_COL_MAJOR, compz_i, n_i, d_i, e_i, z_i,
00176                              ldz_i );
00177 
00178     failed = compare_csteqr( d, d_i, e, e_i, z, z_i, info, info_i, compz, ldz,
00179                              n );
00180     if( failed == 0 ) {
00181         printf( "PASSED: column-major high-level interface to csteqr\n" );
00182     } else {
00183         printf( "FAILED: column-major high-level interface to csteqr\n" );
00184     }
00185 
00186     /* Initialize input data, call the row-major middle-level
00187      * interface to LAPACK routine and check the results */
00188     for( i = 0; i < n; i++ ) {
00189         d_i[i] = d_save[i];
00190     }
00191     for( i = 0; i < (n-1); i++ ) {
00192         e_i[i] = e_save[i];
00193     }
00194     for( i = 0; i < ldz*n; i++ ) {
00195         z_i[i] = z_save[i];
00196     }
00197     for( i = 0; i < (MAX(1,2*n-2)); i++ ) {
00198         work_i[i] = work[i];
00199     }
00200 
00201     if( LAPACKE_lsame( compz, 'i' ) || LAPACKE_lsame( compz, 'v' ) ) {
00202         LAPACKE_cge_trans( LAPACK_COL_MAJOR, n, n, z_i, ldz, z_r, n+2 );
00203     }
00204     info_i = LAPACKE_csteqr_work( LAPACK_ROW_MAJOR, compz_i, n_i, d_i, e_i, z_r,
00205                                   ldz_r, work_i );
00206 
00207     if( LAPACKE_lsame( compz, 'i' ) || LAPACKE_lsame( compz, 'v' ) ) {
00208         LAPACKE_cge_trans( LAPACK_ROW_MAJOR, n, n, z_r, n+2, z_i, ldz );
00209     }
00210 
00211     failed = compare_csteqr( d, d_i, e, e_i, z, z_i, info, info_i, compz, ldz,
00212                              n );
00213     if( failed == 0 ) {
00214         printf( "PASSED: row-major middle-level interface to csteqr\n" );
00215     } else {
00216         printf( "FAILED: row-major middle-level interface to csteqr\n" );
00217     }
00218 
00219     /* Initialize input data, call the row-major high-level
00220      * interface to LAPACK routine and check the results */
00221     for( i = 0; i < n; i++ ) {
00222         d_i[i] = d_save[i];
00223     }
00224     for( i = 0; i < (n-1); i++ ) {
00225         e_i[i] = e_save[i];
00226     }
00227     for( i = 0; i < ldz*n; i++ ) {
00228         z_i[i] = z_save[i];
00229     }
00230     for( i = 0; i < (MAX(1,2*n-2)); i++ ) {
00231         work_i[i] = work[i];
00232     }
00233 
00234     /* Init row_major arrays */
00235     if( LAPACKE_lsame( compz, 'i' ) || LAPACKE_lsame( compz, 'v' ) ) {
00236         LAPACKE_cge_trans( LAPACK_COL_MAJOR, n, n, z_i, ldz, z_r, n+2 );
00237     }
00238     info_i = LAPACKE_csteqr( LAPACK_ROW_MAJOR, compz_i, n_i, d_i, e_i, z_r,
00239                              ldz_r );
00240 
00241     if( LAPACKE_lsame( compz, 'i' ) || LAPACKE_lsame( compz, 'v' ) ) {
00242         LAPACKE_cge_trans( LAPACK_ROW_MAJOR, n, n, z_r, n+2, z_i, ldz );
00243     }
00244 
00245     failed = compare_csteqr( d, d_i, e, e_i, z, z_i, info, info_i, compz, ldz,
00246                              n );
00247     if( failed == 0 ) {
00248         printf( "PASSED: row-major high-level interface to csteqr\n" );
00249     } else {
00250         printf( "FAILED: row-major high-level interface to csteqr\n" );
00251     }
00252 
00253     /* Release memory */
00254     if( d != NULL ) {
00255         LAPACKE_free( d );
00256     }
00257     if( d_i != NULL ) {
00258         LAPACKE_free( d_i );
00259     }
00260     if( d_save != NULL ) {
00261         LAPACKE_free( d_save );
00262     }
00263     if( e != NULL ) {
00264         LAPACKE_free( e );
00265     }
00266     if( e_i != NULL ) {
00267         LAPACKE_free( e_i );
00268     }
00269     if( e_save != NULL ) {
00270         LAPACKE_free( e_save );
00271     }
00272     if( z != NULL ) {
00273         LAPACKE_free( z );
00274     }
00275     if( z_i != NULL ) {
00276         LAPACKE_free( z_i );
00277     }
00278     if( z_r != NULL ) {
00279         LAPACKE_free( z_r );
00280     }
00281     if( z_save != NULL ) {
00282         LAPACKE_free( z_save );
00283     }
00284     if( work != NULL ) {
00285         LAPACKE_free( work );
00286     }
00287     if( work_i != NULL ) {
00288         LAPACKE_free( work_i );
00289     }
00290 
00291     return 0;
00292 }
00293 
00294 /* Auxiliary function: csteqr scalar parameters initialization */
00295 static void init_scalars_csteqr( char *compz, lapack_int *n, lapack_int *ldz )
00296 {
00297     *compz = 'V';
00298     *n = 4;
00299     *ldz = 8;
00300 
00301     return;
00302 }
00303 
00304 /* Auxiliary functions: csteqr array parameters initialization */
00305 static void init_d( lapack_int size, float *d ) {
00306     lapack_int i;
00307     for( i = 0; i < size; i++ ) {
00308         d[i] = 0;
00309     }
00310     d[0] = -3.130000114e+000;
00311     d[1] = -1.285294175e+000;
00312     d[2] = -9.961823225e-001;
00313     d[3] = -1.998523712e+000;
00314 }
00315 static void init_e( lapack_int size, float *e ) {
00316     lapack_int i;
00317     for( i = 0; i < size; i++ ) {
00318         e[i] = 0;
00319     }
00320     e[0] = 4.449280739e+000;
00321     e[1] = 1.700240731e+000;
00322     e[2] = 2.514369249e+000;
00323 }
00324 static void init_z( lapack_int size, lapack_complex_float *z ) {
00325     lapack_int i;
00326     for( i = 0; i < size; i++ ) {
00327         z[i] = lapack_make_complex_float( 0.0f, 0.0f );
00328     }
00329     z[0] = lapack_make_complex_float( 1.000000000e+000, 0.000000000e+000 );
00330     z[8] = lapack_make_complex_float( 0.000000000e+000, 0.000000000e+000 );
00331     z[16] = lapack_make_complex_float( 0.000000000e+000, 0.000000000e+000 );
00332     z[24] = lapack_make_complex_float( 0.000000000e+000, 0.000000000e+000 );
00333     z[1] = lapack_make_complex_float( 0.000000000e+000, 0.000000000e+000 );
00334     z[9] = lapack_make_complex_float( 4.360255301e-001, 4.719863832e-001 );
00335     z[17] = lapack_make_complex_float( 1.789289564e-001, 2.536888719e-001 );
00336     z[25] = lapack_make_complex_float( -6.192449331e-001, -3.275251985e-001 );
00337     z[2] = lapack_make_complex_float( 0.000000000e+000, 0.000000000e+000 );
00338     z[10] = lapack_make_complex_float( -7.641685009e-001, -5.618888140e-002 );
00339     z[18] = lapack_make_complex_float( 2.548894584e-001, 5.297882855e-002 );
00340     z[26] = lapack_make_complex_float( -5.669454336e-001, 1.539055556e-001 );
00341     z[3] = lapack_make_complex_float( 0.000000000e+000, 0.000000000e+000 );
00342     z[11] = lapack_make_complex_float( 0.000000000e+000, 0.000000000e+000 );
00343     z[19] = lapack_make_complex_float( 8.716895580e-001, -2.756962180e-001 );
00344     z[27] = lapack_make_complex_float( 2.890720665e-001, -2.838773131e-001 );
00345 }
00346 static void init_work( lapack_int size, float *work ) {
00347     lapack_int i;
00348     for( i = 0; i < size; i++ ) {
00349         work[i] = 0;
00350     }
00351 }
00352 
00353 /* Auxiliary function: C interface to csteqr results check */
00354 /* Return value: 0 - test is passed, non-zero - test is failed */
00355 static int compare_csteqr( float *d, float *d_i, float *e, float *e_i,
00356                            lapack_complex_float *z, lapack_complex_float *z_i,
00357                            lapack_int info, lapack_int info_i, char compz,
00358                            lapack_int ldz, lapack_int n )
00359 {
00360     lapack_int i;
00361     int failed = 0;
00362     for( i = 0; i < n; i++ ) {
00363         failed += compare_floats(d[i],d_i[i]);
00364     }
00365     for( i = 0; i < (n-1); i++ ) {
00366         failed += compare_floats(e[i],e_i[i]);
00367     }
00368     if( LAPACKE_lsame( compz, 'i' ) || LAPACKE_lsame( compz, 'v' ) ) {
00369         for( i = 0; i < ldz*n; i++ ) {
00370             failed += compare_complex_floats(z[i],z_i[i]);
00371         }
00372     }
00373     failed += (info == info_i) ? 0 : 1;
00374     if( info != 0 || info_i != 0 ) {
00375         printf( "info=%d, info_i=%d\n",(int)info,(int)info_i );
00376     }
00377 
00378     return failed;
00379 }


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