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

Add cuda sub2ind and unitest.

Browse Source
This commit is contained in:
sunwen
2023-12-12 15:20:26 +08:00
parent bd4a27a17b
commit cca2358825
3 changed files with 84 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

22
test/Function2D_Cuda_Test.cpp
View File

@@ -843,8 +843,8 @@ TEST_F(Function2D_Cuda_Test, prod) {
EXPECT_FLOAT_AE(result1[i], result2[i]);
}
auto matrixHostComplex = Aurora::Matrix::fromRawData(new float[40], 4,5, 1,Aurora::Complex);
for(unsigned int i=0; i<40;++i)
auto matrixHostComplex = Aurora::Matrix::fromRawData(new float[20], 2,5, 1,Aurora::Complex);
for(unsigned int i=0; i<20;++i)
{
matrixHost[i] = i + 1;
}
@@ -858,4 +858,22 @@ TEST_F(Function2D_Cuda_Test, prod) {
{
EXPECT_FLOAT_AE(result1[i], result2[i]);
}
}
TEST_F(Function2D_Cuda_Test, sub2ind) {
float* dI1= new float[4]{1,2,1,2};
Aurora::Matrix I1(std::shared_ptr<float>(dI1,std::default_delete<float[]>()),std::vector<int>{4});
float* dI2= new float[4]{2,2,1,1};
Aurora::Matrix I2(std::shared_ptr<float>(dI2,std::default_delete<float[]>()),std::vector<int>{4});
float* dI4= new float[4]{1,1,2,2};
Aurora::Matrix I3(std::shared_ptr<float>(dI4,std::default_delete<float[]>()),std::vector<int>{4});
float* dsz= new float[3]{2,2,2};
Aurora::Matrix sz(std::shared_ptr<float>(dsz,std::default_delete<float[]>()),std::vector<int>{3});
auto result1 = Aurora::sub2ind(sz, {I1, I2, I3});
auto result2 = Aurora::sub2ind(sz.toDeviceMatrix(), {I1.toDeviceMatrix(), I2.toDeviceMatrix(), I3.toDeviceMatrix()}).toHostMatrix();
EXPECT_FLOAT_EQ(result1.getDataSize(), result2.getDataSize());
for(unsigned int i=0; i<result1.getDataSize(); ++i)
{
EXPECT_FLOAT_EQ(result1[i], result2[i]);
}
}