lapacke_cgels_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 cgels
00030 * Author: Intel Corporation
00031 * Generated October, 2010
00032 *****************************************************************************/
00033 
00034 #include "lapacke.h"
00035 #include "lapacke_utils.h"
00036 
00037 lapack_int LAPACKE_cgels_work( int matrix_order, char trans, lapack_int m,
00038                                lapack_int n, lapack_int nrhs,
00039                                lapack_complex_float* a, lapack_int lda,
00040                                lapack_complex_float* b, lapack_int ldb,
00041                                lapack_complex_float* work, lapack_int lwork )
00042 {
00043     lapack_int info = 0;
00044     if( matrix_order == LAPACK_COL_MAJOR ) {
00045         /* Call LAPACK function and adjust info */
00046         LAPACK_cgels( &trans, &m, &n, &nrhs, a, &lda, b, &ldb, work, &lwork,
00047                       &info );
00048         if( info < 0 ) {
00049             info = info - 1;
00050         }
00051     } else if( matrix_order == LAPACK_ROW_MAJOR ) {
00052         lapack_int lda_t = MAX(1,m);
00053         lapack_int ldb_t = MAX(1,MAX(m,n));
00054         lapack_complex_float* a_t = NULL;
00055         lapack_complex_float* b_t = NULL;
00056         /* Check leading dimension(s) */
00057         if( lda < n ) {
00058             info = -7;
00059             LAPACKE_xerbla( "LAPACKE_cgels_work", info );
00060             return info;
00061         }
00062         if( ldb < nrhs ) {
00063             info = -9;
00064             LAPACKE_xerbla( "LAPACKE_cgels_work", info );
00065             return info;
00066         }
00067         /* Query optimal working array(s) size if requested */
00068         if( lwork == -1 ) {
00069             LAPACK_cgels( &trans, &m, &n, &nrhs, a, &lda_t, b, &ldb_t, work,
00070                           &lwork, &info );
00071             return (info < 0) ? (info - 1) : info;
00072         }
00073         /* Allocate memory for temporary array(s) */
00074         a_t = (lapack_complex_float*)
00075             LAPACKE_malloc( sizeof(lapack_complex_float) * lda_t * MAX(1,n) );
00076         if( a_t == NULL ) {
00077             info = LAPACK_TRANSPOSE_MEMORY_ERROR;
00078             goto exit_level_0;
00079         }
00080         b_t = (lapack_complex_float*)
00081             LAPACKE_malloc( sizeof(lapack_complex_float) *
00082                             ldb_t * MAX(1,nrhs) );
00083         if( b_t == NULL ) {
00084             info = LAPACK_TRANSPOSE_MEMORY_ERROR;
00085             goto exit_level_1;
00086         }
00087         /* Transpose input matrices */
00088         LAPACKE_cge_trans( matrix_order, m, n, a, lda, a_t, lda_t );
00089         LAPACKE_cge_trans( matrix_order, MAX(m,n), nrhs, b, ldb, b_t, ldb_t );
00090         /* Call LAPACK function and adjust info */
00091         LAPACK_cgels( &trans, &m, &n, &nrhs, a_t, &lda_t, b_t, &ldb_t, work,
00092                       &lwork, &info );
00093         if( info < 0 ) {
00094             info = info - 1;
00095         }
00096         /* Transpose output matrices */
00097         LAPACKE_cge_trans( LAPACK_COL_MAJOR, m, n, a_t, lda_t, a, lda );
00098         LAPACKE_cge_trans( LAPACK_COL_MAJOR, MAX(m,n), nrhs, b_t, ldb_t, b,
00099                            ldb );
00100         /* Release memory and exit */
00101         LAPACKE_free( b_t );
00102 exit_level_1:
00103         LAPACKE_free( a_t );
00104 exit_level_0:
00105         if( info == LAPACK_TRANSPOSE_MEMORY_ERROR ) {
00106             LAPACKE_xerbla( "LAPACKE_cgels_work", info );
00107         }
00108     } else {
00109         info = -1;
00110         LAPACKE_xerbla( "LAPACKE_cgels_work", info );
00111     }
00112     return info;
00113 }


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