From 1cb8d62f017e085e653eff16f542ffe096981d35 Mon Sep 17 00:00:00 2001
From: kradchen <chenjuijun@realcan.cn>
Date: Fri, 8 Dec 2023 16:19:11 +0800
Subject: [PATCH] UnitTest for complex mul and div for CudaMatrix

---
 test/CudaMatrix_Test.cpp | 47 ++++++++++++++++++++++++++++++++++++++++
 1 file changed, 47 insertions(+)

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:"<<i;
+    }
+    ret2 = A.toDeviceMatrix()*dB;
+    for (size_t i = 0; i < ret1.getDataSize()*2; i++)
+    {
+        ASSERT_FLOAT_EQ(ret1[i], ret2.getValue(i))<<"index:"<<i;
+    }
+    ret2 = dA*B.toDeviceMatrix();
+    for (size_t i = 0; i < ret1.getDataSize()*2; i++)
+    {
+        ASSERT_FLOAT_EQ(ret1[i], ret2.getValue(i))<<"index:"<<i;
+    }
+    ret1 = A/B;
+    ret2 = dA/dB;
+    for (size_t i = 0; i < ret1.getDataSize()*2; i++)
+    {
+        ASSERT_FLOAT_EQ(ret1[i], ret2.getValue(i))<<"index:"<<i;
+    }
+    ret2 = A.toDeviceMatrix()/dB;
+    for (size_t i = 0; i < ret1.getDataSize()*2; i++)
+    {
+        ASSERT_FLOAT_EQ(ret1[i], ret2.getValue(i))<<"index:"<<i;
+    }
+    ret2 = dA/B.toDeviceMatrix();
+    for (size_t i = 0; i < ret1.getDataSize()*2; i++)
+    {
+        ASSERT_FLOAT_EQ(ret1[i], ret2.getValue(i))<<"index:"<<i;
+    }
+
+}
+
 TEST_F(CudaMatrix_Test, MatrixPow){
     auto A = Aurora::zeros(1000,1,1);
     auto B = Aurora::zeros(1000,1,1);