lapacke_cgbrfs_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 cgbrfs
00030 * Author: Intel Corporation
00031 * Generated October, 2010
00032 *****************************************************************************/
00033 
00034 #include "lapacke.h"
00035 #include "lapacke_utils.h"
00036 
00037 lapack_int LAPACKE_cgbrfs_work( int matrix_order, char trans, lapack_int n,
00038                                 lapack_int kl, lapack_int ku, lapack_int nrhs,
00039                                 const lapack_complex_float* ab, lapack_int ldab,
00040                                 const lapack_complex_float* afb,
00041                                 lapack_int ldafb, const lapack_int* ipiv,
00042                                 const lapack_complex_float* b, lapack_int ldb,
00043                                 lapack_complex_float* x, lapack_int ldx,
00044                                 float* ferr, float* berr,
00045                                 lapack_complex_float* work, float* rwork )
00046 {
00047     lapack_int info = 0;
00048     if( matrix_order == LAPACK_COL_MAJOR ) {
00049         /* Call LAPACK function and adjust info */
00050         LAPACK_cgbrfs( &trans, &n, &kl, &ku, &nrhs, ab, &ldab, afb, &ldafb,
00051                        ipiv, b, &ldb, x, &ldx, ferr, berr, work, rwork, &info );
00052         if( info < 0 ) {
00053             info = info - 1;
00054         }
00055     } else if( matrix_order == LAPACK_ROW_MAJOR ) {
00056         lapack_int ldab_t = MAX(1,kl+ku+1);
00057         lapack_int ldafb_t = MAX(1,2*kl+ku+1);
00058         lapack_int ldb_t = MAX(1,n);
00059         lapack_int ldx_t = MAX(1,n);
00060         lapack_complex_float* ab_t = NULL;
00061         lapack_complex_float* afb_t = NULL;
00062         lapack_complex_float* b_t = NULL;
00063         lapack_complex_float* x_t = NULL;
00064         /* Check leading dimension(s) */
00065         if( ldab < n ) {
00066             info = -8;
00067             LAPACKE_xerbla( "LAPACKE_cgbrfs_work", info );
00068             return info;
00069         }
00070         if( ldafb < n ) {
00071             info = -10;
00072             LAPACKE_xerbla( "LAPACKE_cgbrfs_work", info );
00073             return info;
00074         }
00075         if( ldb < nrhs ) {
00076             info = -13;
00077             LAPACKE_xerbla( "LAPACKE_cgbrfs_work", info );
00078             return info;
00079         }
00080         if( ldx < nrhs ) {
00081             info = -15;
00082             LAPACKE_xerbla( "LAPACKE_cgbrfs_work", info );
00083             return info;
00084         }
00085         /* Allocate memory for temporary array(s) */
00086         ab_t = (lapack_complex_float*)
00087             LAPACKE_malloc( sizeof(lapack_complex_float) * ldab_t * MAX(1,n) );
00088         if( ab_t == NULL ) {
00089             info = LAPACK_TRANSPOSE_MEMORY_ERROR;
00090             goto exit_level_0;
00091         }
00092         afb_t = (lapack_complex_float*)
00093             LAPACKE_malloc( sizeof(lapack_complex_float) * ldafb_t * MAX(1,n) );
00094         if( afb_t == NULL ) {
00095             info = LAPACK_TRANSPOSE_MEMORY_ERROR;
00096             goto exit_level_1;
00097         }
00098         b_t = (lapack_complex_float*)
00099             LAPACKE_malloc( sizeof(lapack_complex_float) *
00100                             ldb_t * MAX(1,nrhs) );
00101         if( b_t == NULL ) {
00102             info = LAPACK_TRANSPOSE_MEMORY_ERROR;
00103             goto exit_level_2;
00104         }
00105         x_t = (lapack_complex_float*)
00106             LAPACKE_malloc( sizeof(lapack_complex_float) *
00107                             ldx_t * MAX(1,nrhs) );
00108         if( x_t == NULL ) {
00109             info = LAPACK_TRANSPOSE_MEMORY_ERROR;
00110             goto exit_level_3;
00111         }
00112         /* Transpose input matrices */
00113         LAPACKE_cgb_trans( matrix_order, n, n, kl, ku, ab, ldab, ab_t, ldab_t );
00114         LAPACKE_cgb_trans( matrix_order, n, n, kl, kl+ku, afb, ldafb, afb_t,
00115                            ldafb_t );
00116         LAPACKE_cge_trans( matrix_order, n, nrhs, b, ldb, b_t, ldb_t );
00117         LAPACKE_cge_trans( matrix_order, n, nrhs, x, ldx, x_t, ldx_t );
00118         /* Call LAPACK function and adjust info */
00119         LAPACK_cgbrfs( &trans, &n, &kl, &ku, &nrhs, ab_t, &ldab_t, afb_t,
00120                        &ldafb_t, ipiv, b_t, &ldb_t, x_t, &ldx_t, ferr, berr,
00121                        work, rwork, &info );
00122         if( info < 0 ) {
00123             info = info - 1;
00124         }
00125         /* Transpose output matrices */
00126         LAPACKE_cge_trans( LAPACK_COL_MAJOR, n, nrhs, x_t, ldx_t, x, ldx );
00127         /* Release memory and exit */
00128         LAPACKE_free( x_t );
00129 exit_level_3:
00130         LAPACKE_free( b_t );
00131 exit_level_2:
00132         LAPACKE_free( afb_t );
00133 exit_level_1:
00134         LAPACKE_free( ab_t );
00135 exit_level_0:
00136         if( info == LAPACK_TRANSPOSE_MEMORY_ERROR ) {
00137             LAPACKE_xerbla( "LAPACKE_cgbrfs_work", info );
00138         }
00139     } else {
00140         info = -1;
00141         LAPACKE_xerbla( "LAPACKE_cgbrfs_work", info );
00142     }
00143     return info;
00144 }


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