diff --git a/test/CudaMatrix_Test.cpp b/test/CudaMatrix_Test.cpp index 7e894d0..ac82609 100644 --- a/test/CudaMatrix_Test.cpp +++ b/test/CudaMatrix_Test.cpp @@ -350,6 +350,53 @@ TEST_F(CudaMatrix_Test, MatrixDiv) { } } +TEST_F(CudaMatrix_Test, MatrixComplexMulAndDiv){ + float * dataA = new float[100]; + float * dataB = new float[100]; + auto A = Aurora::Matrix::fromRawData(dataA,5,10,1,Aurora::Complex); + auto B = Aurora::Matrix::fromRawData(dataB,5,10,1,Aurora::Complex); + for (size_t i = 0; i < 100; i++) + { + dataA[i] = (float)1; + dataB[i] = (float)(-2); + } + auto dA = A.toDeviceMatrix(); + auto dB = B.toDeviceMatrix(); + auto ret1 = A*B; + auto ret2 = dA*dB; + for (size_t i = 0; i < ret1.getDataSize()*2; i++) + { + ASSERT_FLOAT_EQ(ret1[i], ret2.getValue(i))<<"index:"<