sgehrd_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 * sgehrd_1 is the test program for the C interface to LAPACK
00036 * routine sgehrd
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_sgehrd( lapack_int *n, lapack_int *ilo,
00055                                  lapack_int *ihi, lapack_int *lda,
00056                                  lapack_int *lwork );
00057 static void init_a( lapack_int size, float *a );
00058 static void init_tau( lapack_int size, float *tau );
00059 static void init_work( lapack_int size, float *work );
00060 static int compare_sgehrd( float *a, float *a_i, float *tau, float *tau_i,
00061                            lapack_int info, lapack_int info_i, lapack_int lda,
00062                            lapack_int n );
00063 
00064 int main(void)
00065 {
00066     /* Local scalars */
00067     lapack_int n, n_i;
00068     lapack_int ilo, ilo_i;
00069     lapack_int ihi, ihi_i;
00070     lapack_int lda, lda_i;
00071     lapack_int lda_r;
00072     lapack_int lwork, lwork_i;
00073     lapack_int info, info_i;
00074     lapack_int i;
00075     int failed;
00076 
00077     /* Local arrays */
00078     float *a = NULL, *a_i = NULL;
00079     float *tau = NULL, *tau_i = NULL;
00080     float *work = NULL, *work_i = NULL;
00081     float *a_save = NULL;
00082     float *tau_save = NULL;
00083     float *a_r = NULL;
00084 
00085     /* Iniitialize the scalar parameters */
00086     init_scalars_sgehrd( &n, &ilo, &ihi, &lda, &lwork );
00087     lda_r = n+2;
00088     n_i = n;
00089     ilo_i = ilo;
00090     ihi_i = ihi;
00091     lda_i = lda;
00092     lwork_i = lwork;
00093 
00094     /* Allocate memory for the LAPACK routine arrays */
00095     a = (float *)LAPACKE_malloc( lda*n * sizeof(float) );
00096     tau = (float *)LAPACKE_malloc( (n-1) * sizeof(float) );
00097     work = (float *)LAPACKE_malloc( lwork * sizeof(float) );
00098 
00099     /* Allocate memory for the C interface function arrays */
00100     a_i = (float *)LAPACKE_malloc( lda*n * sizeof(float) );
00101     tau_i = (float *)LAPACKE_malloc( (n-1) * sizeof(float) );
00102     work_i = (float *)LAPACKE_malloc( lwork * sizeof(float) );
00103 
00104     /* Allocate memory for the backup arrays */
00105     a_save = (float *)LAPACKE_malloc( lda*n * sizeof(float) );
00106     tau_save = (float *)LAPACKE_malloc( (n-1) * sizeof(float) );
00107 
00108     /* Allocate memory for the row-major arrays */
00109     a_r = (float *)LAPACKE_malloc( n*(n+2) * sizeof(float) );
00110 
00111     /* Initialize input arrays */
00112     init_a( lda*n, a );
00113     init_tau( (n-1), tau );
00114     init_work( lwork, work );
00115 
00116     /* Backup the ouptut arrays */
00117     for( i = 0; i < lda*n; i++ ) {
00118         a_save[i] = a[i];
00119     }
00120     for( i = 0; i < (n-1); i++ ) {
00121         tau_save[i] = tau[i];
00122     }
00123 
00124     /* Call the LAPACK routine */
00125     sgehrd_( &n, &ilo, &ihi, a, &lda, tau, work, &lwork, &info );
00126 
00127     /* Initialize input data, call the column-major middle-level
00128      * interface to LAPACK routine and check the results */
00129     for( i = 0; i < lda*n; i++ ) {
00130         a_i[i] = a_save[i];
00131     }
00132     for( i = 0; i < (n-1); i++ ) {
00133         tau_i[i] = tau_save[i];
00134     }
00135     for( i = 0; i < lwork; i++ ) {
00136         work_i[i] = work[i];
00137     }
00138     info_i = LAPACKE_sgehrd_work( LAPACK_COL_MAJOR, n_i, ilo_i, ihi_i, a_i,
00139                                   lda_i, tau_i, work_i, lwork_i );
00140 
00141     failed = compare_sgehrd( a, a_i, tau, tau_i, info, info_i, lda, n );
00142     if( failed == 0 ) {
00143         printf( "PASSED: column-major middle-level interface to sgehrd\n" );
00144     } else {
00145         printf( "FAILED: column-major middle-level interface to sgehrd\n" );
00146     }
00147 
00148     /* Initialize input data, call the column-major high-level
00149      * interface to LAPACK routine and check the results */
00150     for( i = 0; i < lda*n; i++ ) {
00151         a_i[i] = a_save[i];
00152     }
00153     for( i = 0; i < (n-1); i++ ) {
00154         tau_i[i] = tau_save[i];
00155     }
00156     for( i = 0; i < lwork; i++ ) {
00157         work_i[i] = work[i];
00158     }
00159     info_i = LAPACKE_sgehrd( LAPACK_COL_MAJOR, n_i, ilo_i, ihi_i, a_i, lda_i,
00160                              tau_i );
00161 
00162     failed = compare_sgehrd( a, a_i, tau, tau_i, info, info_i, lda, n );
00163     if( failed == 0 ) {
00164         printf( "PASSED: column-major high-level interface to sgehrd\n" );
00165     } else {
00166         printf( "FAILED: column-major high-level interface to sgehrd\n" );
00167     }
00168 
00169     /* Initialize input data, call the row-major middle-level
00170      * interface to LAPACK routine and check the results */
00171     for( i = 0; i < lda*n; i++ ) {
00172         a_i[i] = a_save[i];
00173     }
00174     for( i = 0; i < (n-1); i++ ) {
00175         tau_i[i] = tau_save[i];
00176     }
00177     for( i = 0; i < lwork; i++ ) {
00178         work_i[i] = work[i];
00179     }
00180 
00181     LAPACKE_sge_trans( LAPACK_COL_MAJOR, n, n, a_i, lda, a_r, n+2 );
00182     info_i = LAPACKE_sgehrd_work( LAPACK_ROW_MAJOR, n_i, ilo_i, ihi_i, a_r,
00183                                   lda_r, tau_i, work_i, lwork_i );
00184 
00185     LAPACKE_sge_trans( LAPACK_ROW_MAJOR, n, n, a_r, n+2, a_i, lda );
00186 
00187     failed = compare_sgehrd( a, a_i, tau, tau_i, info, info_i, lda, n );
00188     if( failed == 0 ) {
00189         printf( "PASSED: row-major middle-level interface to sgehrd\n" );
00190     } else {
00191         printf( "FAILED: row-major middle-level interface to sgehrd\n" );
00192     }
00193 
00194     /* Initialize input data, call the row-major high-level
00195      * interface to LAPACK routine and check the results */
00196     for( i = 0; i < lda*n; i++ ) {
00197         a_i[i] = a_save[i];
00198     }
00199     for( i = 0; i < (n-1); i++ ) {
00200         tau_i[i] = tau_save[i];
00201     }
00202     for( i = 0; i < lwork; i++ ) {
00203         work_i[i] = work[i];
00204     }
00205 
00206     /* Init row_major arrays */
00207     LAPACKE_sge_trans( LAPACK_COL_MAJOR, n, n, a_i, lda, a_r, n+2 );
00208     info_i = LAPACKE_sgehrd( LAPACK_ROW_MAJOR, n_i, ilo_i, ihi_i, a_r, lda_r,
00209                              tau_i );
00210 
00211     LAPACKE_sge_trans( LAPACK_ROW_MAJOR, n, n, a_r, n+2, a_i, lda );
00212 
00213     failed = compare_sgehrd( a, a_i, tau, tau_i, info, info_i, lda, n );
00214     if( failed == 0 ) {
00215         printf( "PASSED: row-major high-level interface to sgehrd\n" );
00216     } else {
00217         printf( "FAILED: row-major high-level interface to sgehrd\n" );
00218     }
00219 
00220     /* Release memory */
00221     if( a != NULL ) {
00222         LAPACKE_free( a );
00223     }
00224     if( a_i != NULL ) {
00225         LAPACKE_free( a_i );
00226     }
00227     if( a_r != NULL ) {
00228         LAPACKE_free( a_r );
00229     }
00230     if( a_save != NULL ) {
00231         LAPACKE_free( a_save );
00232     }
00233     if( tau != NULL ) {
00234         LAPACKE_free( tau );
00235     }
00236     if( tau_i != NULL ) {
00237         LAPACKE_free( tau_i );
00238     }
00239     if( tau_save != NULL ) {
00240         LAPACKE_free( tau_save );
00241     }
00242     if( work != NULL ) {
00243         LAPACKE_free( work );
00244     }
00245     if( work_i != NULL ) {
00246         LAPACKE_free( work_i );
00247     }
00248 
00249     return 0;
00250 }
00251 
00252 /* Auxiliary function: sgehrd scalar parameters initialization */
00253 static void init_scalars_sgehrd( lapack_int *n, lapack_int *ilo,
00254                                  lapack_int *ihi, lapack_int *lda,
00255                                  lapack_int *lwork )
00256 {
00257     *n = 4;
00258     *ilo = 2;
00259     *ihi = 4;
00260     *lda = 8;
00261     *lwork = 512;
00262 
00263     return;
00264 }
00265 
00266 /* Auxiliary functions: sgehrd array parameters initialization */
00267 static void init_a( lapack_int size, float *a ) {
00268     lapack_int i;
00269     for( i = 0; i < size; i++ ) {
00270         a[i] = 0;
00271     }
00272     a[0] = -4.000000060e-001;  /* a[0,0] */
00273     a[8] = 3.200000048e+000;  /* a[0,1] */
00274     a[16] = 7.599999905e+000;  /* a[0,2] */
00275     a[24] = -1.500000000e+000;  /* a[0,3] */
00276     a[1] = 0.000000000e+000;  /* a[1,0] */
00277     a[9] = 2.000000030e-001;  /* a[1,1] */
00278     a[17] = 9.100000262e-001;  /* a[1,2] */
00279     a[25] = 4.312500000e+000;  /* a[1,3] */
00280     a[2] = 0.000000000e+000;  /* a[2,0] */
00281     a[10] = 9.100000262e-001;  /* a[2,1] */
00282     a[18] = 5.139999866e+000;  /* a[2,2] */
00283     a[26] = -4.099999905e+000;  /* a[2,3] */
00284     a[3] = 0.000000000e+000;  /* a[3,0] */
00285     a[11] = 2.799999952e+000;  /* a[3,1] */
00286     a[19] = -2.640000105e+000;  /* a[3,2] */
00287     a[27] = 6.600000262e-001;  /* a[3,3] */
00288 }
00289 static void init_tau( lapack_int size, float *tau ) {
00290     lapack_int i;
00291     for( i = 0; i < size; i++ ) {
00292         tau[i] = 0;
00293     }
00294 }
00295 static void init_work( lapack_int size, float *work ) {
00296     lapack_int i;
00297     for( i = 0; i < size; i++ ) {
00298         work[i] = 0;
00299     }
00300 }
00301 
00302 /* Auxiliary function: C interface to sgehrd results check */
00303 /* Return value: 0 - test is passed, non-zero - test is failed */
00304 static int compare_sgehrd( float *a, float *a_i, float *tau, float *tau_i,
00305                            lapack_int info, lapack_int info_i, lapack_int lda,
00306                            lapack_int n )
00307 {
00308     lapack_int i;
00309     int failed = 0;
00310     for( i = 0; i < lda*n; i++ ) {
00311         failed += compare_floats(a[i],a_i[i]);
00312     }
00313     for( i = 0; i < (n-1); i++ ) {
00314         failed += compare_floats(tau[i],tau_i[i]);
00315     }
00316     failed += (info == info_i) ? 0 : 1;
00317     if( info != 0 || info_i != 0 ) {
00318         printf( "info=%d, info_i=%d\n",(int)info,(int)info_i );
00319     }
00320 
00321     return failed;
00322 }


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