Add cuda linspace and unittest.

2023-12-01 16:15:51 +08:00
parent 7db741502e
commit 65c78cd878
3 changed files with 32 additions and 0 deletions
--- a/test/Function1D_Cuda_Test.cpp
+++ b/test/Function1D_Cuda_Test.cpp
@@ -940,3 +940,13 @@ TEST_F(Function1D_Cuda_Test, vecnorm) {
    EXPECT_FLOAT_AE(result.getData()[0],9.5394);
    EXPECT_FLOAT_AE(result.getData()[1],43.3474);
 }
+
+TEST_F(Function1D_Cuda_Test, linspace) {
+    auto result1 = Aurora::linspace(-5,5,7);
+    auto result2 = Aurora::linspaceCuda(-5,5,7).toHostMatrix();
+    EXPECT_FLOAT_EQ(result1.getDataSize(), result2.getDataSize());
+    for(int i=0; i<result1.getDataSize(); ++i)
+    {
+        EXPECT_FLOAT_AE(result1[i], result2[i]);
+    }
+}