lapacke_shgeqz_work.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: Native middle-level C interface to LAPACK function shgeqz
00030 * Author: Intel Corporation
00031 * Generated October, 2010
00032 *****************************************************************************/
00033 
00034 #include "lapacke.h"
00035 #include "lapacke_utils.h"
00036 
00037 lapack_int LAPACKE_shgeqz_work( int matrix_order, char job, char compq,
00038                                 char compz, lapack_int n, lapack_int ilo,
00039                                 lapack_int ihi, float* h, lapack_int ldh,
00040                                 float* t, lapack_int ldt, float* alphar,
00041                                 float* alphai, float* beta, float* q,
00042                                 lapack_int ldq, float* z, lapack_int ldz,
00043                                 float* work, lapack_int lwork )
00044 {
00045     lapack_int info = 0;
00046     if( matrix_order == LAPACK_COL_MAJOR ) {
00047         /* Call LAPACK function and adjust info */
00048         LAPACK_shgeqz( &job, &compq, &compz, &n, &ilo, &ihi, h, &ldh, t, &ldt,
00049                        alphar, alphai, beta, q, &ldq, z, &ldz, work, &lwork,
00050                        &info );
00051         if( info < 0 ) {
00052             info = info - 1;
00053         }
00054     } else if( matrix_order == LAPACK_ROW_MAJOR ) {
00055         lapack_int ldh_t = MAX(1,n);
00056         lapack_int ldq_t = MAX(1,n);
00057         lapack_int ldt_t = MAX(1,n);
00058         lapack_int ldz_t = MAX(1,n);
00059         float* h_t = NULL;
00060         float* t_t = NULL;
00061         float* q_t = NULL;
00062         float* z_t = NULL;
00063         /* Check leading dimension(s) */
00064         if( ldh < n ) {
00065             info = -9;
00066             LAPACKE_xerbla( "LAPACKE_shgeqz_work", info );
00067             return info;
00068         }
00069         if( ldq < n ) {
00070             info = -16;
00071             LAPACKE_xerbla( "LAPACKE_shgeqz_work", info );
00072             return info;
00073         }
00074         if( ldt < n ) {
00075             info = -11;
00076             LAPACKE_xerbla( "LAPACKE_shgeqz_work", info );
00077             return info;
00078         }
00079         if( ldz < n ) {
00080             info = -18;
00081             LAPACKE_xerbla( "LAPACKE_shgeqz_work", info );
00082             return info;
00083         }
00084         /* Query optimal working array(s) size if requested */
00085         if( lwork == -1 ) {
00086             LAPACK_shgeqz( &job, &compq, &compz, &n, &ilo, &ihi, h, &ldh_t, t,
00087                            &ldt_t, alphar, alphai, beta, q, &ldq_t, z, &ldz_t,
00088                            work, &lwork, &info );
00089             return (info < 0) ? (info - 1) : info;
00090         }
00091         /* Allocate memory for temporary array(s) */
00092         h_t = (float*)LAPACKE_malloc( sizeof(float) * ldh_t * MAX(1,n) );
00093         if( h_t == NULL ) {
00094             info = LAPACK_TRANSPOSE_MEMORY_ERROR;
00095             goto exit_level_0;
00096         }
00097         t_t = (float*)LAPACKE_malloc( sizeof(float) * ldt_t * MAX(1,n) );
00098         if( t_t == NULL ) {
00099             info = LAPACK_TRANSPOSE_MEMORY_ERROR;
00100             goto exit_level_1;
00101         }
00102         if( LAPACKE_lsame( compq, 'i' ) || LAPACKE_lsame( compq, 'v' ) ) {
00103             q_t = (float*)LAPACKE_malloc( sizeof(float) * ldq_t * MAX(1,n) );
00104             if( q_t == NULL ) {
00105                 info = LAPACK_TRANSPOSE_MEMORY_ERROR;
00106                 goto exit_level_2;
00107             }
00108         }
00109         if( LAPACKE_lsame( compz, 'i' ) || LAPACKE_lsame( compz, 'v' ) ) {
00110             z_t = (float*)LAPACKE_malloc( sizeof(float) * ldz_t * MAX(1,n) );
00111             if( z_t == NULL ) {
00112                 info = LAPACK_TRANSPOSE_MEMORY_ERROR;
00113                 goto exit_level_3;
00114             }
00115         }
00116         /* Transpose input matrices */
00117         LAPACKE_sge_trans( matrix_order, n, n, h, ldh, h_t, ldh_t );
00118         LAPACKE_sge_trans( matrix_order, n, n, t, ldt, t_t, ldt_t );
00119         if( LAPACKE_lsame( compq, 'v' ) ) {
00120             LAPACKE_sge_trans( matrix_order, n, n, q, ldq, q_t, ldq_t );
00121         }
00122         if( LAPACKE_lsame( compz, 'v' ) ) {
00123             LAPACKE_sge_trans( matrix_order, n, n, z, ldz, z_t, ldz_t );
00124         }
00125         /* Call LAPACK function and adjust info */
00126         LAPACK_shgeqz( &job, &compq, &compz, &n, &ilo, &ihi, h_t, &ldh_t, t_t,
00127                        &ldt_t, alphar, alphai, beta, q_t, &ldq_t, z_t, &ldz_t,
00128                        work, &lwork, &info );
00129         if( info < 0 ) {
00130             info = info - 1;
00131         }
00132         /* Transpose output matrices */
00133         LAPACKE_sge_trans( LAPACK_COL_MAJOR, n, n, h_t, ldh_t, h, ldh );
00134         LAPACKE_sge_trans( LAPACK_COL_MAJOR, n, n, t_t, ldt_t, t, ldt );
00135         if( LAPACKE_lsame( compq, 'i' ) || LAPACKE_lsame( compq, 'v' ) ) {
00136             LAPACKE_sge_trans( LAPACK_COL_MAJOR, n, n, q_t, ldq_t, q, ldq );
00137         }
00138         if( LAPACKE_lsame( compz, 'i' ) || LAPACKE_lsame( compz, 'v' ) ) {
00139             LAPACKE_sge_trans( LAPACK_COL_MAJOR, n, n, z_t, ldz_t, z, ldz );
00140         }
00141         /* Release memory and exit */
00142         if( LAPACKE_lsame( compz, 'i' ) || LAPACKE_lsame( compz, 'v' ) ) {
00143             LAPACKE_free( z_t );
00144         }
00145 exit_level_3:
00146         if( LAPACKE_lsame( compq, 'i' ) || LAPACKE_lsame( compq, 'v' ) ) {
00147             LAPACKE_free( q_t );
00148         }
00149 exit_level_2:
00150         LAPACKE_free( t_t );
00151 exit_level_1:
00152         LAPACKE_free( h_t );
00153 exit_level_0:
00154         if( info == LAPACK_TRANSPOSE_MEMORY_ERROR ) {
00155             LAPACKE_xerbla( "LAPACKE_shgeqz_work", info );
00156         }
00157     } else {
00158         info = -1;
00159         LAPACKE_xerbla( "LAPACKE_shgeqz_work", info );
00160     }
00161     return info;
00162 }


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