lapacke_ztrrfs_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 ztrrfs
00030 * Author: Intel Corporation
00031 * Generated October, 2010
00032 *****************************************************************************/
00033 
00034 #include "lapacke.h"
00035 #include "lapacke_utils.h"
00036 
00037 lapack_int LAPACKE_ztrrfs_work( int matrix_order, char uplo, char trans,
00038                                 char diag, lapack_int n, lapack_int nrhs,
00039                                 const lapack_complex_double* a, lapack_int lda,
00040                                 const lapack_complex_double* b, lapack_int ldb,
00041                                 const lapack_complex_double* x, lapack_int ldx,
00042                                 double* ferr, double* berr,
00043                                 lapack_complex_double* work, double* rwork )
00044 {
00045     lapack_int info = 0;
00046     if( matrix_order == LAPACK_COL_MAJOR ) {
00047         /* Call LAPACK function and adjust info */
00048         LAPACK_ztrrfs( &uplo, &trans, &diag, &n, &nrhs, a, &lda, b, &ldb, x,
00049                        &ldx, ferr, berr, work, rwork, &info );
00050         if( info < 0 ) {
00051             info = info - 1;
00052         }
00053     } else if( matrix_order == LAPACK_ROW_MAJOR ) {
00054         lapack_int lda_t = MAX(1,n);
00055         lapack_int ldb_t = MAX(1,n);
00056         lapack_int ldx_t = MAX(1,n);
00057         lapack_complex_double* a_t = NULL;
00058         lapack_complex_double* b_t = NULL;
00059         lapack_complex_double* x_t = NULL;
00060         /* Check leading dimension(s) */
00061         if( lda < n ) {
00062             info = -8;
00063             LAPACKE_xerbla( "LAPACKE_ztrrfs_work", info );
00064             return info;
00065         }
00066         if( ldb < nrhs ) {
00067             info = -10;
00068             LAPACKE_xerbla( "LAPACKE_ztrrfs_work", info );
00069             return info;
00070         }
00071         if( ldx < nrhs ) {
00072             info = -12;
00073             LAPACKE_xerbla( "LAPACKE_ztrrfs_work", info );
00074             return info;
00075         }
00076         /* Allocate memory for temporary array(s) */
00077         a_t = (lapack_complex_double*)
00078             LAPACKE_malloc( sizeof(lapack_complex_double) * lda_t * MAX(1,n) );
00079         if( a_t == NULL ) {
00080             info = LAPACK_TRANSPOSE_MEMORY_ERROR;
00081             goto exit_level_0;
00082         }
00083         b_t = (lapack_complex_double*)
00084             LAPACKE_malloc( sizeof(lapack_complex_double) *
00085                             ldb_t * MAX(1,nrhs) );
00086         if( b_t == NULL ) {
00087             info = LAPACK_TRANSPOSE_MEMORY_ERROR;
00088             goto exit_level_1;
00089         }
00090         x_t = (lapack_complex_double*)
00091             LAPACKE_malloc( sizeof(lapack_complex_double) *
00092                             ldx_t * MAX(1,nrhs) );
00093         if( x_t == NULL ) {
00094             info = LAPACK_TRANSPOSE_MEMORY_ERROR;
00095             goto exit_level_2;
00096         }
00097         /* Transpose input matrices */
00098         LAPACKE_ztr_trans( matrix_order, uplo, diag, n, a, lda, a_t, lda_t );
00099         LAPACKE_zge_trans( matrix_order, n, nrhs, b, ldb, b_t, ldb_t );
00100         LAPACKE_zge_trans( matrix_order, n, nrhs, x, ldx, x_t, ldx_t );
00101         /* Call LAPACK function and adjust info */
00102         LAPACK_ztrrfs( &uplo, &trans, &diag, &n, &nrhs, a_t, &lda_t, b_t,
00103                        &ldb_t, x_t, &ldx_t, ferr, berr, work, rwork, &info );
00104         if( info < 0 ) {
00105             info = info - 1;
00106         }
00107         /* Release memory and exit */
00108         LAPACKE_free( x_t );
00109 exit_level_2:
00110         LAPACKE_free( b_t );
00111 exit_level_1:
00112         LAPACKE_free( a_t );
00113 exit_level_0:
00114         if( info == LAPACK_TRANSPOSE_MEMORY_ERROR ) {
00115             LAPACKE_xerbla( "LAPACKE_ztrrfs_work", info );
00116         }
00117     } else {
00118         info = -1;
00119         LAPACKE_xerbla( "LAPACKE_ztrrfs_work", info );
00120     }
00121     return info;
00122 }


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