cunmqr_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 * cunmqr_1 is the test program for the C interface to LAPACK
00036 * routine cunmqr
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_cunmqr( char *side, char *trans, lapack_int *m,
00055                                  lapack_int *n, lapack_int *k, lapack_int *lda,
00056                                  lapack_int *ldc, lapack_int *lwork );
00057 static void init_a( lapack_int size, lapack_complex_float *a );
00058 static void init_tau( lapack_int size, lapack_complex_float *tau );
00059 static void init_c( lapack_int size, lapack_complex_float *c );
00060 static void init_work( lapack_int size, lapack_complex_float *work );
00061 static int compare_cunmqr( lapack_complex_float *c, lapack_complex_float *c_i,
00062                            lapack_int info, lapack_int info_i, lapack_int ldc,
00063                            lapack_int n );
00064 
00065 int main(void)
00066 {
00067     /* Local scalars */
00068     char side, side_i;
00069     char trans, trans_i;
00070     lapack_int m, m_i;
00071     lapack_int n, n_i;
00072     lapack_int k, k_i;
00073     lapack_int lda, lda_i;
00074     lapack_int lda_r;
00075     lapack_int ldc, ldc_i;
00076     lapack_int ldc_r;
00077     lapack_int lwork, lwork_i;
00078     lapack_int info, info_i;
00079     /* Declare scalars */
00080     lapack_int r;
00081     lapack_int i;
00082     int failed;
00083 
00084     /* Local arrays */
00085     lapack_complex_float *a = NULL, *a_i = NULL;
00086     lapack_complex_float *tau = NULL, *tau_i = NULL;
00087     lapack_complex_float *c = NULL, *c_i = NULL;
00088     lapack_complex_float *work = NULL, *work_i = NULL;
00089     lapack_complex_float *c_save = NULL;
00090     lapack_complex_float *a_r = NULL;
00091     lapack_complex_float *c_r = NULL;
00092 
00093     /* Iniitialize the scalar parameters */
00094     init_scalars_cunmqr( &side, &trans, &m, &n, &k, &lda, &ldc, &lwork );
00095     r = LAPACKE_lsame( side, 'l' ) ? m : n;
00096     lda_r = k+2;
00097     ldc_r = n+2;
00098     side_i = side;
00099     trans_i = trans;
00100     m_i = m;
00101     n_i = n;
00102     k_i = k;
00103     lda_i = lda;
00104     ldc_i = ldc;
00105     lwork_i = lwork;
00106 
00107     /* Allocate memory for the LAPACK routine arrays */
00108     a = (lapack_complex_float *)
00109         LAPACKE_malloc( lda*k * sizeof(lapack_complex_float) );
00110     tau = (lapack_complex_float *)
00111         LAPACKE_malloc( k * sizeof(lapack_complex_float) );
00112     c = (lapack_complex_float *)
00113         LAPACKE_malloc( ldc*n * sizeof(lapack_complex_float) );
00114     work = (lapack_complex_float *)
00115         LAPACKE_malloc( lwork * sizeof(lapack_complex_float) );
00116 
00117     /* Allocate memory for the C interface function arrays */
00118     a_i = (lapack_complex_float *)
00119         LAPACKE_malloc( lda*k * sizeof(lapack_complex_float) );
00120     tau_i = (lapack_complex_float *)
00121         LAPACKE_malloc( k * sizeof(lapack_complex_float) );
00122     c_i = (lapack_complex_float *)
00123         LAPACKE_malloc( ldc*n * sizeof(lapack_complex_float) );
00124     work_i = (lapack_complex_float *)
00125         LAPACKE_malloc( lwork * sizeof(lapack_complex_float) );
00126 
00127     /* Allocate memory for the backup arrays */
00128     c_save = (lapack_complex_float *)
00129         LAPACKE_malloc( ldc*n * sizeof(lapack_complex_float) );
00130 
00131     /* Allocate memory for the row-major arrays */
00132     a_r = (lapack_complex_float *)
00133         LAPACKE_malloc( r*(k+2) * sizeof(lapack_complex_float) );
00134     c_r = (lapack_complex_float *)
00135         LAPACKE_malloc( m*(n+2) * sizeof(lapack_complex_float) );
00136 
00137     /* Initialize input arrays */
00138     init_a( lda*k, a );
00139     init_tau( k, tau );
00140     init_c( ldc*n, c );
00141     init_work( lwork, work );
00142 
00143     /* Backup the ouptut arrays */
00144     for( i = 0; i < ldc*n; i++ ) {
00145         c_save[i] = c[i];
00146     }
00147 
00148     /* Call the LAPACK routine */
00149     cunmqr_( &side, &trans, &m, &n, &k, a, &lda, tau, c, &ldc, work, &lwork,
00150              &info );
00151 
00152     /* Initialize input data, call the column-major middle-level
00153      * interface to LAPACK routine and check the results */
00154     for( i = 0; i < lda*k; i++ ) {
00155         a_i[i] = a[i];
00156     }
00157     for( i = 0; i < k; i++ ) {
00158         tau_i[i] = tau[i];
00159     }
00160     for( i = 0; i < ldc*n; i++ ) {
00161         c_i[i] = c_save[i];
00162     }
00163     for( i = 0; i < lwork; i++ ) {
00164         work_i[i] = work[i];
00165     }
00166     info_i = LAPACKE_cunmqr_work( LAPACK_COL_MAJOR, side_i, trans_i, m_i, n_i,
00167                                   k_i, a_i, lda_i, tau_i, c_i, ldc_i, work_i,
00168                                   lwork_i );
00169 
00170     failed = compare_cunmqr( c, c_i, info, info_i, ldc, n );
00171     if( failed == 0 ) {
00172         printf( "PASSED: column-major middle-level interface to cunmqr\n" );
00173     } else {
00174         printf( "FAILED: column-major middle-level interface to cunmqr\n" );
00175     }
00176 
00177     /* Initialize input data, call the column-major high-level
00178      * interface to LAPACK routine and check the results */
00179     for( i = 0; i < lda*k; i++ ) {
00180         a_i[i] = a[i];
00181     }
00182     for( i = 0; i < k; i++ ) {
00183         tau_i[i] = tau[i];
00184     }
00185     for( i = 0; i < ldc*n; i++ ) {
00186         c_i[i] = c_save[i];
00187     }
00188     for( i = 0; i < lwork; i++ ) {
00189         work_i[i] = work[i];
00190     }
00191     info_i = LAPACKE_cunmqr( LAPACK_COL_MAJOR, side_i, trans_i, m_i, n_i, k_i,
00192                              a_i, lda_i, tau_i, c_i, ldc_i );
00193 
00194     failed = compare_cunmqr( c, c_i, info, info_i, ldc, n );
00195     if( failed == 0 ) {
00196         printf( "PASSED: column-major high-level interface to cunmqr\n" );
00197     } else {
00198         printf( "FAILED: column-major high-level interface to cunmqr\n" );
00199     }
00200 
00201     /* Initialize input data, call the row-major middle-level
00202      * interface to LAPACK routine and check the results */
00203     for( i = 0; i < lda*k; i++ ) {
00204         a_i[i] = a[i];
00205     }
00206     for( i = 0; i < k; i++ ) {
00207         tau_i[i] = tau[i];
00208     }
00209     for( i = 0; i < ldc*n; i++ ) {
00210         c_i[i] = c_save[i];
00211     }
00212     for( i = 0; i < lwork; i++ ) {
00213         work_i[i] = work[i];
00214     }
00215 
00216     LAPACKE_cge_trans( LAPACK_COL_MAJOR, r, k, a_i, lda, a_r, k+2 );
00217     LAPACKE_cge_trans( LAPACK_COL_MAJOR, m, n, c_i, ldc, c_r, n+2 );
00218     info_i = LAPACKE_cunmqr_work( LAPACK_ROW_MAJOR, side_i, trans_i, m_i, n_i,
00219                                   k_i, a_r, lda_r, tau_i, c_r, ldc_r, work_i,
00220                                   lwork_i );
00221 
00222     LAPACKE_cge_trans( LAPACK_ROW_MAJOR, m, n, c_r, n+2, c_i, ldc );
00223 
00224     failed = compare_cunmqr( c, c_i, info, info_i, ldc, n );
00225     if( failed == 0 ) {
00226         printf( "PASSED: row-major middle-level interface to cunmqr\n" );
00227     } else {
00228         printf( "FAILED: row-major middle-level interface to cunmqr\n" );
00229     }
00230 
00231     /* Initialize input data, call the row-major high-level
00232      * interface to LAPACK routine and check the results */
00233     for( i = 0; i < lda*k; i++ ) {
00234         a_i[i] = a[i];
00235     }
00236     for( i = 0; i < k; i++ ) {
00237         tau_i[i] = tau[i];
00238     }
00239     for( i = 0; i < ldc*n; i++ ) {
00240         c_i[i] = c_save[i];
00241     }
00242     for( i = 0; i < lwork; i++ ) {
00243         work_i[i] = work[i];
00244     }
00245 
00246     /* Init row_major arrays */
00247     LAPACKE_cge_trans( LAPACK_COL_MAJOR, r, k, a_i, lda, a_r, k+2 );
00248     LAPACKE_cge_trans( LAPACK_COL_MAJOR, m, n, c_i, ldc, c_r, n+2 );
00249     info_i = LAPACKE_cunmqr( LAPACK_ROW_MAJOR, side_i, trans_i, m_i, n_i, k_i,
00250                              a_r, lda_r, tau_i, c_r, ldc_r );
00251 
00252     LAPACKE_cge_trans( LAPACK_ROW_MAJOR, m, n, c_r, n+2, c_i, ldc );
00253 
00254     failed = compare_cunmqr( c, c_i, info, info_i, ldc, n );
00255     if( failed == 0 ) {
00256         printf( "PASSED: row-major high-level interface to cunmqr\n" );
00257     } else {
00258         printf( "FAILED: row-major high-level interface to cunmqr\n" );
00259     }
00260 
00261     /* Release memory */
00262     if( a != NULL ) {
00263         LAPACKE_free( a );
00264     }
00265     if( a_i != NULL ) {
00266         LAPACKE_free( a_i );
00267     }
00268     if( a_r != NULL ) {
00269         LAPACKE_free( a_r );
00270     }
00271     if( tau != NULL ) {
00272         LAPACKE_free( tau );
00273     }
00274     if( tau_i != NULL ) {
00275         LAPACKE_free( tau_i );
00276     }
00277     if( c != NULL ) {
00278         LAPACKE_free( c );
00279     }
00280     if( c_i != NULL ) {
00281         LAPACKE_free( c_i );
00282     }
00283     if( c_r != NULL ) {
00284         LAPACKE_free( c_r );
00285     }
00286     if( c_save != NULL ) {
00287         LAPACKE_free( c_save );
00288     }
00289     if( work != NULL ) {
00290         LAPACKE_free( work );
00291     }
00292     if( work_i != NULL ) {
00293         LAPACKE_free( work_i );
00294     }
00295 
00296     return 0;
00297 }
00298 
00299 /* Auxiliary function: cunmqr scalar parameters initialization */
00300 static void init_scalars_cunmqr( char *side, char *trans, lapack_int *m,
00301                                  lapack_int *n, lapack_int *k, lapack_int *lda,
00302                                  lapack_int *ldc, lapack_int *lwork )
00303 {
00304     *side = 'L';
00305     *trans = 'C';
00306     *m = 5;
00307     *n = 2;
00308     *k = 3;
00309     *lda = 8;
00310     *ldc = 8;
00311     *lwork = 512;
00312 
00313     return;
00314 }
00315 
00316 /* Auxiliary functions: cunmqr array parameters initialization */
00317 static void init_a( lapack_int size, lapack_complex_float *a ) {
00318     lapack_int i;
00319     for( i = 0; i < size; i++ ) {
00320         a[i] = lapack_make_complex_float( 0.0f, 0.0f );
00321     }
00322     a[0] = lapack_make_complex_float( -2.287006617e+000, 0.000000000e+000 );
00323     a[8] = lapack_make_complex_float( -7.238719463e-001, 5.605143309e-001 );
00324     a[16] = lapack_make_complex_float( 7.569720745e-001, 4.519444704e-001 );
00325     a[1] = lapack_make_complex_float( -1.419878304e-001, 8.154310286e-003 );
00326     a[9] = lapack_make_complex_float( 1.895002127e+000, 0.000000000e+000 );
00327     a[17] = lapack_make_complex_float( -3.620421290e-001, -4.786572456e-001 );
00328     a[2] = lapack_make_complex_float( -9.061318636e-002, -5.909461156e-002 );
00329     a[10] = lapack_make_complex_float( -4.734396636e-001, 1.684395820e-001 );
00330     a[18] = lapack_make_complex_float( 9.392829537e-001, 0.000000000e+000 );
00331     a[3] = lapack_make_complex_float( 4.082103670e-001, -1.177922338e-001 );
00332     a[11] = lapack_make_complex_float( 2.585218847e-001, 2.667200565e-002 );
00333     a[19] = lapack_make_complex_float( 4.509717971e-002, 5.815137625e-001 );
00334     a[4] = lapack_make_complex_float( -8.956561983e-002, 3.396957815e-001 );
00335     a[12] = lapack_make_complex_float( -6.871830225e-001, 3.170874715e-002 );
00336     a[20] = lapack_make_complex_float( -1.170649379e-001, -1.206572726e-002 );
00337 }
00338 static void init_tau( lapack_int size, lapack_complex_float *tau ) {
00339     lapack_int i;
00340     for( i = 0; i < size; i++ ) {
00341         tau[i] = lapack_make_complex_float( 0.0f, 0.0f );
00342     }
00343     tau[0] = lapack_make_complex_float( 1.349802256e+000, -4.459978342e-001 );
00344     tau[1] = lapack_make_complex_float( 1.040377498e+000, 2.791351080e-001 );
00345     tau[2] = lapack_make_complex_float( 1.470288873e+000, 9.977108985e-002 );
00346 }
00347 static void init_c( lapack_int size, lapack_complex_float *c ) {
00348     lapack_int i;
00349     for( i = 0; i < size; i++ ) {
00350         c[i] = lapack_make_complex_float( 0.0f, 0.0f );
00351     }
00352     c[0] = lapack_make_complex_float( -8.500000238e-001, -1.629999995e+000 );
00353     c[8] = lapack_make_complex_float( 2.490000010e+000, 4.010000229e+000 );
00354     c[1] = lapack_make_complex_float( -2.160000086e+000, 3.519999981e+000 );
00355     c[9] = lapack_make_complex_float( -1.400000006e-001, 7.980000019e+000 );
00356     c[2] = lapack_make_complex_float( 4.570000172e+000, -5.710000038e+000 );
00357     c[10] = lapack_make_complex_float( 8.359999657e+000, -2.800000012e-001 );
00358     c[3] = lapack_make_complex_float( 6.380000114e+000, -7.400000095e+000 );
00359     c[11] = lapack_make_complex_float( -3.549999952e+000, 1.289999962e+000 );
00360     c[4] = lapack_make_complex_float( 8.409999847e+000, 9.390000343e+000 );
00361     c[12] = lapack_make_complex_float( -6.719999790e+000, 5.030000210e+000 );
00362 }
00363 static void init_work( lapack_int size, lapack_complex_float *work ) {
00364     lapack_int i;
00365     for( i = 0; i < size; i++ ) {
00366         work[i] = lapack_make_complex_float( 0.0f, 0.0f );
00367     }
00368 }
00369 
00370 /* Auxiliary function: C interface to cunmqr results check */
00371 /* Return value: 0 - test is passed, non-zero - test is failed */
00372 static int compare_cunmqr( lapack_complex_float *c, lapack_complex_float *c_i,
00373                            lapack_int info, lapack_int info_i, lapack_int ldc,
00374                            lapack_int n )
00375 {
00376     lapack_int i;
00377     int failed = 0;
00378     for( i = 0; i < ldc*n; i++ ) {
00379         failed += compare_complex_floats(c[i],c_i[i]);
00380     }
00381     failed += (info == info_i) ? 0 : 1;
00382     if( info != 0 || info_i != 0 ) {
00383         printf( "info=%d, info_i=%d\n",(int)info,(int)info_i );
00384     }
00385 
00386     return failed;
00387 }


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