sunwen/Aurora
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Add cuda auroraUnion and unittest.

Browse Source
This commit is contained in:
sunwen
2023-12-04 11:27:17 +08:00
parent 65c78cd878
commit 5e77b4dafe
3 changed files with 38 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
test/Function1D_Cuda_Test.cpp
View File

@@ -950,3 +950,18 @@ TEST_F(Function1D_Cuda_Test, linspace) {
EXPECT_FLOAT_AE(result1[i], result2[i]);
}
}
TEST_F(Function1D_Cuda_Test, auroraUnion) {
float* data1 = new float[9]{3,3,2,2,2,1,4,4,7};
auto matrix1 = Aurora::Matrix::fromRawData(data1, 9,1,1);
float* data2 = new float[8]{6,6,7,7,8,1,2};
auto matrix2 = Aurora::Matrix::fromRawData(data2, 7,1,1);
auto result1 = Aurora::auroraUnion(matrix1, matrix2);
auto result2 = Aurora::auroraUnion(matrix1.toDeviceMatrix(), matrix2.toDeviceMatrix()).toHostMatrix();
EXPECT_FLOAT_AE(result1.getDataSize(), result2.getDataSize());
for(int i=0;i<result1.getDataSize();++i)
{
EXPECT_FLOAT_AE(result1[i], result2[i]);
}
}