#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_her2(int64_t n,
T * alpha,
T *X,
T *Y,
T *A )
{
obj_t obj_a;
obj_t obj_x, obj_y;
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;
bli_obj_create_with_attached_buffer(dt, 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_create_with_attached_buffer( dt, n, 1, Y, 1, n,&obj_y );
bli_obj_set_struc( BLIS_HERMITIAN, &obj_a );
bli_obj_set_uplo( BLIS_LOWER, &obj_a);
bli_her2( &obj_alpha,
&obj_x,
&obj_y,
&obj_a );
}
template< typename T >
void test_her2( )
{
T *A, *X, *Y, *A_ref;
T alpha;
int n;
int lda, incx, incy, lda_ref;
alpha = ALPHA;
n = N;
lda = n;
lda_ref = n;
incx = 1;
incy = 1;
srand (time(NULL));
allocate_init_buffer(A , n , n);
allocate_init_buffer(X , n , 1);
allocate_init_buffer(Y , n , 1);
copy_buffer(A, A_ref , n ,n);
#ifdef PRINT
printmatrix(A, lda ,n,n,(char *) "A");
printvector(X, n,(char *) "X");
printvector(Y, n, (char *)"Y");
#endif
blis::her2(
CblasColMajor,
CblasLower,
n,
alpha,
X,
incx,
Y,
incy,
A,
lda
);
#ifdef PRINT
printmatrix(A, lda , n , n,(char *) "A output");
#endif
ref_her2(n, &alpha, X, Y, A_ref);
#ifdef PRINT
printmatrix(A_ref, lda , n, n, (char *)"A_ref output");
#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[]( Y );
delete[]( A_ref );
}
int main( int argc, char** argv )
{
test_her2<complex<float>>( );
test_her2<complex<double>>( );
return 0;
}