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

Add deleteColumn and createCudaVectorMatrix.

Browse Source
This commit is contained in:
sunwen
2023-12-05 14:37:00 +08:00
parent 00a21a37ef
commit 6867dcb227
3 changed files with 69 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
test/Function1D_Cuda_Test.cpp
View File

@@ -1103,3 +1103,21 @@ TEST_F(Function1D_Cuda_Test, xcorr) {
EXPECT_FLOAT_AE(result[5], 48);
EXPECT_FLOAT_AE(result[6], 20);
}
TEST_F(Function1D_Cuda_Test, deleteColumn) {
auto matrixHost = Aurora::Matrix::fromRawData(new float[100], 10,10,1);
for(int i=0;i<100;++i)
{
matrixHost[i] = i;
}
auto matrixDevice = matrixHost.toDeviceMatrix();
auto result1 = Aurora::deleteColumn(matrixHost, 26);
auto result2 = Aurora::deleteColumn(matrixDevice, 26).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());
for(int i=0;i<result1.getDataSize();++i)
{
EXPECT_FLOAT_AE(result1[i], result2[i]);
}
}