#include #include "CudaMatrix.h" #include "Matrix.h" #include "TestUtility.h" #include "Function1D.h" #include "Function1D.cuh" class Function1D_Cuda_Test:public ::testing::Test { protected: static void SetUpFunction1DCudaTester(){ } static void TearDownTestCase(){ } void SetUp(){ } void TearDown(){ } }; TEST_F(Function1D_Cuda_Test, complex) { Aurora::Matrix hostMatrix = Aurora::Matrix::fromRawData(new float[8]{1,2,3,4,5,6,7,8}, 2,2,2); Aurora::CudaMatrix deviceMatrix = hostMatrix.toDeviceMatrix(); auto result1 = Aurora::complex(hostMatrix); auto result2 = Aurora::complex(deviceMatrix).toHostMatrix(); EXPECT_EQ(result2.getDataSize(), 8); EXPECT_EQ(result2.getValueType(), Aurora::Complex); for(size_t i=0; i