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_zhfrk_work( int matrix_order, char transr, char uplo,
00038 char trans, lapack_int n, lapack_int k,
00039 double alpha, const lapack_complex_double* a,
00040 lapack_int lda, double beta,
00041 lapack_complex_double* c )
00042 {
00043 lapack_int info = 0;
00044 if( matrix_order == LAPACK_COL_MAJOR ) {
00045
00046 LAPACK_zhfrk( &transr, &uplo, &trans, &n, &k, &alpha, a, &lda, &beta,
00047 c );
00048 if( info < 0 ) {
00049 info = info - 1;
00050 }
00051 } else if( matrix_order == LAPACK_ROW_MAJOR ) {
00052 lapack_int na = LAPACKE_lsame( trans, 'n' ) ? n : k;
00053 lapack_int ka = LAPACKE_lsame( trans, 'n' ) ? k : n;
00054 lapack_int lda_t = MAX(1,na);
00055 lapack_complex_double* a_t = NULL;
00056 lapack_complex_double* c_t = NULL;
00057
00058 if( lda < ka ) {
00059 info = -9;
00060 LAPACKE_xerbla( "LAPACKE_zhfrk_work", info );
00061 return info;
00062 }
00063
00064 a_t = (lapack_complex_double*)
00065 LAPACKE_malloc( sizeof(lapack_complex_double) * lda_t * MAX(1,ka) );
00066 if( a_t == NULL ) {
00067 info = LAPACK_TRANSPOSE_MEMORY_ERROR;
00068 goto exit_level_0;
00069 }
00070 c_t = (lapack_complex_double*)
00071 LAPACKE_malloc( sizeof(lapack_complex_double) *
00072 ( MAX(1,n) * MAX(2,n+1) ) / 2 );
00073 if( c_t == NULL ) {
00074 info = LAPACK_TRANSPOSE_MEMORY_ERROR;
00075 goto exit_level_1;
00076 }
00077
00078 LAPACKE_zge_trans( matrix_order, na, ka, a, lda, a_t, lda_t );
00079 LAPACKE_zpf_trans( matrix_order, transr, uplo, n, c, c_t );
00080
00081 LAPACK_zhfrk( &transr, &uplo, &trans, &n, &k, &alpha, a_t, &lda_t,
00082 &beta, c_t );
00083 info = 0;
00084
00085 LAPACKE_zpf_trans( LAPACK_COL_MAJOR, transr, uplo, n, c_t, c );
00086
00087 LAPACKE_free( c_t );
00088 exit_level_1:
00089 LAPACKE_free( a_t );
00090 exit_level_0:
00091 if( info == LAPACK_TRANSPOSE_MEMORY_ERROR ) {
00092 LAPACKE_xerbla( "LAPACKE_zhfrk_work", info );
00093 }
00094 } else {
00095 info = -1;
00096 LAPACKE_xerbla( "LAPACKE_zhfrk_work", info );
00097 }
00098 return info;
00099 }