00001
00002
00003
00004
00005
00006
00007
00008
00009
00010
00011
00012
00013
00014
00015
00016
00017
00018
00019
00020
00021
00022
00023
00024
00025
00026
00027
00028
00029
00030
00031
00032
00033
00034 #include "lapacke.h"
00035 #include "lapacke_utils.h"
00036
00037 lapack_int LAPACKE_chpsv_work( int matrix_order, char uplo, lapack_int n,
00038 lapack_int nrhs, lapack_complex_float* ap,
00039 lapack_int* ipiv, lapack_complex_float* b,
00040 lapack_int ldb )
00041 {
00042 lapack_int info = 0;
00043 if( matrix_order == LAPACK_COL_MAJOR ) {
00044
00045 LAPACK_chpsv( &uplo, &n, &nrhs, ap, ipiv, b, &ldb, &info );
00046 if( info < 0 ) {
00047 info = info - 1;
00048 }
00049 } else if( matrix_order == LAPACK_ROW_MAJOR ) {
00050 lapack_int ldb_t = MAX(1,n);
00051 lapack_complex_float* b_t = NULL;
00052 lapack_complex_float* ap_t = NULL;
00053
00054 if( ldb < nrhs ) {
00055 info = -8;
00056 LAPACKE_xerbla( "LAPACKE_chpsv_work", info );
00057 return info;
00058 }
00059
00060 b_t = (lapack_complex_float*)
00061 LAPACKE_malloc( sizeof(lapack_complex_float) *
00062 ldb_t * MAX(1,nrhs) );
00063 if( b_t == NULL ) {
00064 info = LAPACK_TRANSPOSE_MEMORY_ERROR;
00065 goto exit_level_0;
00066 }
00067 ap_t = (lapack_complex_float*)
00068 LAPACKE_malloc( sizeof(lapack_complex_float) *
00069 ( MAX(1,n) * MAX(2,n+1) ) / 2 );
00070 if( ap_t == NULL ) {
00071 info = LAPACK_TRANSPOSE_MEMORY_ERROR;
00072 goto exit_level_1;
00073 }
00074
00075 LAPACKE_cge_trans( matrix_order, n, nrhs, b, ldb, b_t, ldb_t );
00076 LAPACKE_chp_trans( matrix_order, uplo, n, ap, ap_t );
00077
00078 LAPACK_chpsv( &uplo, &n, &nrhs, ap_t, ipiv, b_t, &ldb_t, &info );
00079 if( info < 0 ) {
00080 info = info - 1;
00081 }
00082
00083 LAPACKE_cge_trans( LAPACK_COL_MAJOR, n, nrhs, b_t, ldb_t, b, ldb );
00084 LAPACKE_chp_trans( LAPACK_COL_MAJOR, uplo, n, ap_t, ap );
00085
00086 LAPACKE_free( ap_t );
00087 exit_level_1:
00088 LAPACKE_free( b_t );
00089 exit_level_0:
00090 if( info == LAPACK_TRANSPOSE_MEMORY_ERROR ) {
00091 LAPACKE_xerbla( "LAPACKE_chpsv_work", info );
00092 }
00093 } else {
00094 info = -1;
00095 LAPACKE_xerbla( "LAPACKE_chpsv_work", info );
00096 }
00097 return info;
00098 }