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 zgttrs 00030 * Author: Intel Corporation 00031 * Generated October, 2010 00032 *****************************************************************************/ 00033 00034 #include "lapacke.h" 00035 #include "lapacke_utils.h" 00036 00037 lapack_int LAPACKE_zgttrs_work( int matrix_order, char trans, lapack_int n, 00038 lapack_int nrhs, 00039 const lapack_complex_double* dl, 00040 const lapack_complex_double* d, 00041 const lapack_complex_double* du, 00042 const lapack_complex_double* du2, 00043 const lapack_int* ipiv, 00044 lapack_complex_double* b, lapack_int ldb ) 00045 { 00046 lapack_int info = 0; 00047 if( matrix_order == LAPACK_COL_MAJOR ) { 00048 /* Call LAPACK function and adjust info */ 00049 LAPACK_zgttrs( &trans, &n, &nrhs, dl, d, du, du2, ipiv, b, &ldb, 00050 &info ); 00051 if( info < 0 ) { 00052 info = info - 1; 00053 } 00054 } else if( matrix_order == LAPACK_ROW_MAJOR ) { 00055 lapack_int ldb_t = MAX(1,n); 00056 lapack_complex_double* b_t = NULL; 00057 /* Check leading dimension(s) */ 00058 if( ldb < nrhs ) { 00059 info = -11; 00060 LAPACKE_xerbla( "LAPACKE_zgttrs_work", info ); 00061 return info; 00062 } 00063 /* Allocate memory for temporary array(s) */ 00064 b_t = (lapack_complex_double*) 00065 LAPACKE_malloc( sizeof(lapack_complex_double) * 00066 ldb_t * MAX(1,nrhs) ); 00067 if( b_t == NULL ) { 00068 info = LAPACK_TRANSPOSE_MEMORY_ERROR; 00069 goto exit_level_0; 00070 } 00071 /* Transpose input matrices */ 00072 LAPACKE_zge_trans( matrix_order, n, nrhs, b, ldb, b_t, ldb_t ); 00073 /* Call LAPACK function and adjust info */ 00074 LAPACK_zgttrs( &trans, &n, &nrhs, dl, d, du, du2, ipiv, b_t, &ldb_t, 00075 &info ); 00076 if( info < 0 ) { 00077 info = info - 1; 00078 } 00079 /* Transpose output matrices */ 00080 LAPACKE_zge_trans( LAPACK_COL_MAJOR, n, nrhs, b_t, ldb_t, b, ldb ); 00081 /* Release memory and exit */ 00082 LAPACKE_free( b_t ); 00083 exit_level_0: 00084 if( info == LAPACK_TRANSPOSE_MEMORY_ERROR ) { 00085 LAPACKE_xerbla( "LAPACKE_zgttrs_work", info ); 00086 } 00087 } else { 00088 info = -1; 00089 LAPACKE_xerbla( "LAPACKE_zgttrs_work", info ); 00090 } 00091 return info; 00092 }