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

Add cuda uniqueByRows and unitest.

Browse Source
This commit is contained in:
sunwen
2023-12-06 15:21:56 +08:00
parent 6867dcb227
commit 6f32a05ee0
3 changed files with 140 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

28
test/Function1D_Cuda_Test.cpp
View File

@@ -1121,3 +1121,31 @@ TEST_F(Function1D_Cuda_Test, deleteColumn) {
EXPECT_FLOAT_AE(result1[i], result2[i]);
}
}
TEST_F(Function1D_Cuda_Test, uniqueByRows) {
auto matrixHost = Aurora::Matrix::fromRawData(new float[12]{6,2,2,2,4,7,7,7,1,8,9,8},4,3,1,Aurora::Normal);
Aurora::Matrix indexHost;
auto matrixDevice = matrixHost.toDeviceMatrix();
Aurora::CudaMatrix indexDevice;
auto result1 = Aurora::uniqueByRows(matrixHost, indexHost);
auto result2 = Aurora::uniqueByRows(matrixDevice, indexDevice).toHostMatrix();
auto indexResult1 = indexHost;
auto indexResult2 = indexDevice.toHostMatrix();
EXPECT_FLOAT_AE(result1.getDimSize(0),result2.getDimSize(0));
EXPECT_FLOAT_AE(result1.getDimSize(1),result2.getDimSize(1));
EXPECT_FLOAT_AE(result1.getDataSize(),result2.getDataSize());
EXPECT_FLOAT_AE(indexResult1.getDimSize(0),indexResult2.getDimSize(0));
EXPECT_FLOAT_AE(indexResult1.getDimSize(1),indexResult2.getDimSize(1));
EXPECT_FLOAT_AE(indexResult1.getDataSize(),indexResult2.getDataSize());
for(int i=0;i<result1.getDataSize();++i)
{
EXPECT_FLOAT_AE(result1[i], result2[i]);
}
for(int i=0;i<result1.getDataSize();++i)
{
EXPECT_FLOAT_AE(indexResult1[i], indexResult2[i]);
}
}