#include <complex>
#include <iostream>
#include "blis.hh"
#include "test.hh"
using namespace blis;
using namespace std;
#define ALPHA 1.0
#define BETA 0.0
#define N 6
template< typename T >
void ref_hemv(int64_t n,
T * alpha,
T *A,
T *X,
T * beta,
T *Y )
{
obj_t obj_a, obj_x, obj_y;
obj_t obj_alpha, obj_beta;
num_t dt;
if(is_same<T, float>::value)
dt = BLIS_FLOAT;
else if(is_same<T, double>::value)
dt = BLIS_DOUBLE;
else 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, 1, 1, beta, 1,1,&obj_beta );
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_hemv( &obj_alpha,
&obj_a,
&obj_x,
&obj_beta,
&obj_y );
}
template< typename T >
void test_hemv( )
{
T *A, *Y, *Y_ref, *X;
T alpha, beta;
int n;
int lda, incx, incy, incy_ref;
alpha = ALPHA;
beta = BETA;
n = N;
lda = n;
incx = 1;
incy = 1;
incy_ref = 1;
srand (time(NULL));
allocate_init_buffer(A , n , n);
allocate_init_buffer(X , n , 1);
allocate_init_buffer(Y , n , 1);
copy_buffer(Y, Y_ref , n ,1);
#ifdef PRINT
printmatrix(A, lda ,n,n, (char *)"A");
printvector(X, n, (char *)"X");
printvector(Y, n, (char *)"Y");
#endif
blis::hemv(
CblasColMajor,
CblasLower,
n,
alpha,
A,
lda,
X,
incx,
beta,
Y,
incy
);
#ifdef PRINT
printvector(Y, n, (char *)"Y output");
#endif
ref_hemv(n, &alpha, A, X, &beta, Y_ref);
#ifdef PRINT
printvector(Y_ref, n,(char *) "Y_ref output");
#endif
if(computeErrorV(incy,incy_ref, n, Y, Y_ref )==1)
printf("%s TEST FAIL\n" , __PRETTY_FUNCTION__ );
else
printf("%s TEST PASS\n" , __PRETTY_FUNCTION__ );
delete[]( A );
delete[]( X );
delete[]( Y );
delete[]( Y_ref );
}
int main( int argc, char** argv )
{
test_hemv<complex<float>>( );
test_hemv<complex<double>>( );
return 0;
}