#include <complex>
#include <iostream>
#include "blis.hh"
#include "test.hh"
using namespace blis;
using namespace std;
#define ALPHA 1.0
#define N 6
template< typename T >
void ref_her(int64_t n,
real_type<T> * alpha,
T *X,
T *A )
{
obj_t obj_a;
obj_t obj_x;
obj_t obj_alpha;
num_t dt;
if(is_same<T, complex<float>>::value)
dt = BLIS_SCOMPLEX;
else if(is_same<T, complex<double>>::value)
dt = BLIS_DCOMPLEX;
if(dt == BLIS_SCOMPLEX){
bli_obj_create_with_attached_buffer( BLIS_FLOAT, 1, 1, alpha, 1,1,&obj_alpha );
}
else if(dt == BLIS_DCOMPLEX){
bli_obj_create_with_attached_buffer( BLIS_DOUBLE, 1, 1, alpha, 1,1,&obj_alpha );
}
bli_obj_create_with_attached_buffer( dt, n, n, A, 1, n, &obj_a );
bli_obj_create_with_attached_buffer( dt, n, 1, X, 1, n,&obj_x );
bli_obj_set_struc( BLIS_HERMITIAN, &obj_a );
bli_obj_set_uplo( BLIS_LOWER, &obj_a);
bli_her( &obj_alpha,
&obj_x,
&obj_a );
}
template< typename T >
void test_her( )
{
T *A, *X, *A_ref;
real_type<T> alpha;
int n;
int lda, incx, lda_ref;
alpha = ALPHA;
n = N;
lda = n;
lda_ref = n;
incx = 1;
srand (time(NULL));
allocate_init_buffer(A , n , n);
allocate_init_buffer(X , n , 1);
copy_buffer(A, A_ref , n ,n);
#ifdef PRINT
printmatrix(A, lda ,n,n,(char *) "A");
printvector(X, n,(char *) "X");
#endif
blis::her(
CblasColMajor,
CblasLower,
n,
alpha,
X,
incx,
A,
lda
);
#ifdef PRINT
printmatrix(A, lda ,n,n, (char *)"A output");
#endif
ref_her(n, &alpha, X, A_ref);
#ifdef PRINT
printmatrix(A_ref, lda_ref, n,n ,(char *) "A refoutput");
#endif
if(computeErrorM(lda, lda_ref, n, n, A, A_ref )==1)
printf("%s TEST FAIL\n" ,__PRETTY_FUNCTION__);
else
printf("%s TEST PASS\n" , __PRETTY_FUNCTION__);
delete[]( A );
delete[]( X );
delete[]( A_ref );
}
int main( int argc, char** argv )
{
test_her<complex<float>>( );
test_her<complex<double>>( );
return 0;
}