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

Fix UnitTest add cudamatrix add and mul

Browse Source
This commit is contained in:
kradchen
2023-11-01 14:31:29 +08:00
parent fe0abf8ee6
commit 029b86013e
13 changed files with 1108 additions and 617 deletions
Download Patch File Download Diff File Expand all files Collapse all files

104
test/Function3D_Test.cpp
View File

@@ -24,73 +24,73 @@ protected:
};
TEST_F(Function3D_Test, interp3) {
double* xD= new double[4]{1,2,3,4};
Aurora::Matrix x(std::shared_ptr<double>(xD,std::default_delete<double[]>()),std::vector<int>{4});
double* yD= new double[5]{1,4,9,16,25};
Aurora::Matrix y(std::shared_ptr<double>(yD,std::default_delete<double[]>()),std::vector<int>{5});
double* zD= new double[3]{22,33,56};
Aurora::Matrix z(std::shared_ptr<double>(zD,std::default_delete<double[]>()),std::vector<int>{3});
double* x1D= new double[3]{1.22,2.69,3.84};
Aurora::Matrix x1(std::shared_ptr<double>(x1D,std::default_delete<double[]>()),std::vector<int>{3});
double* y1D= new double[3]{4.5823,10.552,18};
Aurora::Matrix y1(std::shared_ptr<double>(y1D,std::default_delete<double[]>()),std::vector<int>{3});
double* z1D= new double[3]{22.256,40.22,48};
Aurora::Matrix z1(std::shared_ptr<double>(z1D,std::default_delete<double[]>()),std::vector<int>{3});
double* vD= new double[60]{158,256,332,428,22,56,487,54,26,333,541,554,445,48,22,54,36,98,11,25,
float* xD= new float[4]{1,2,3,4};
Aurora::Matrix x(std::shared_ptr<float>(xD,std::default_delete<float[]>()),std::vector<int>{4});
float* yD= new float[5]{1,4,9,16,25};
Aurora::Matrix y(std::shared_ptr<float>(yD,std::default_delete<float[]>()),std::vector<int>{5});
float* zD= new float[3]{22,33,56};
Aurora::Matrix z(std::shared_ptr<float>(zD,std::default_delete<float[]>()),std::vector<int>{3});
float* x1D= new float[3]{1.22,2.69,3.84};
Aurora::Matrix x1(std::shared_ptr<float>(x1D,std::default_delete<float[]>()),std::vector<int>{3});
float* y1D= new float[3]{4.5823,10.552,18};
Aurora::Matrix y1(std::shared_ptr<float>(y1D,std::default_delete<float[]>()),std::vector<int>{3});
float* z1D= new float[3]{22.256,40.22,48};
Aurora::Matrix z1(std::shared_ptr<float>(z1D,std::default_delete<float[]>()),std::vector<int>{3});
float* vD= new float[60]{158,256,332,428,22,56,487,54,26,333,541,554,445,48,22,54,36,98,11,25,
158,256,332,428,22,56,487,54,26,333,541,554,445,48,22,54,36,98,11,25,
158,256,332,428,22,56,487,54,26,333,541,554,445,48,22,54,36,98,11,25};
Aurora::Matrix v(std::shared_ptr<double>(vD,std::default_delete<double[]>()),std::vector<int>{5,4,3});
Aurora::Matrix v(std::shared_ptr<float>(vD,std::default_delete<float[]>()),std::vector<int>{5,4,3});
Aurora::Matrix result = interp3(x,y,z,v,x1,y1,z1,Aurora::InterpnMethod::Spline);
EXPECT_DOUBLE_AE(result.getData()[0],303.5663);
EXPECT_DOUBLE_AE(result.getData()[1],243.6711);
EXPECT_DOUBLE_AE(result.getData()[2],-47.32);
EXPECT_FLOAT_AE(result.getData()[0],303.5663);
EXPECT_FLOAT_AE(result.getData()[1],243.6711);
EXPECT_FLOAT_AE(result.getData()[2],-47.32);
result = interp3(x,y,z,v,x1,y1,z1,Aurora::InterpnMethod::Linear);
EXPECT_DOUBLE_AE(result.getData()[0],302.6298);
EXPECT_DOUBLE_AE(result.getData()[1],261.1313);
EXPECT_DOUBLE_AE(result.getData()[2],18.6089);
EXPECT_FLOAT_AE(result.getData()[0],302.6298);
EXPECT_FLOAT_AE(result.getData()[1],261.1313);
EXPECT_FLOAT_AE(result.getData()[2],18.6089);
}
TEST_F(Function3D_Test, interpn) {
double* xD= new double[4]{1,2,3,4};
Aurora::Matrix x(std::shared_ptr<double>(xD,std::default_delete<double[]>()),std::vector<int>{4});
double* yD= new double[5]{1,4,9,16,25};
Aurora::Matrix y(std::shared_ptr<double>(yD,std::default_delete<double[]>()),std::vector<int>{5});
double* zD= new double[3]{22,33,56};
Aurora::Matrix z(std::shared_ptr<double>(zD,std::default_delete<double[]>()),std::vector<int>{3});
double* x1D= new double[3]{1.22,2.69,3.84};
Aurora::Matrix x1(std::shared_ptr<double>(x1D,std::default_delete<double[]>()),std::vector<int>{3});
double* y1D= new double[3]{2.5823,3.552,3.888};
Aurora::Matrix y1(std::shared_ptr<double>(y1D,std::default_delete<double[]>()),std::vector<int>{3});
double* z1D= new double[3]{22.256,40.22,48};
Aurora::Matrix z1(std::shared_ptr<double>(z1D,std::default_delete<double[]>()),std::vector<int>{3});
double* v2D= new double[20]{158,256,332,428,22,56,487,54,26,333,541,554,445,48,22,54,36,98,11,25};
Aurora::Matrix v2(std::shared_ptr<double>(v2D,std::default_delete<double[]>()),std::vector<int>{5,4});
double* v3D= new double[60]{158,256,332,428,22,56,487,54,26,333,541,554,445,48,22,54,36,98,11,25,
float* xD= new float[4]{1,2,3,4};
Aurora::Matrix x(std::shared_ptr<float>(xD,std::default_delete<float[]>()),std::vector<int>{4});
float* yD= new float[5]{1,4,9,16,25};
Aurora::Matrix y(std::shared_ptr<float>(yD,std::default_delete<float[]>()),std::vector<int>{5});
float* zD= new float[3]{22,33,56};
Aurora::Matrix z(std::shared_ptr<float>(zD,std::default_delete<float[]>()),std::vector<int>{3});
float* x1D= new float[3]{1.22,2.69,3.84};
Aurora::Matrix x1(std::shared_ptr<float>(x1D,std::default_delete<float[]>()),std::vector<int>{3});
float* y1D= new float[3]{2.5823,3.552,3.888};
Aurora::Matrix y1(std::shared_ptr<float>(y1D,std::default_delete<float[]>()),std::vector<int>{3});
float* z1D= new float[3]{22.256,40.22,48};
Aurora::Matrix z1(std::shared_ptr<float>(z1D,std::default_delete<float[]>()),std::vector<int>{3});
float* v2D= new float[20]{158,256,332,428,22,56,487,54,26,333,541,554,445,48,22,54,36,98,11,25};
Aurora::Matrix v2(std::shared_ptr<float>(v2D,std::default_delete<float[]>()),std::vector<int>{5,4});
float* v3D= new float[60]{158,256,332,428,22,56,487,54,26,333,541,554,445,48,22,54,36,98,11,25,
158,256,332,428,22,56,487,54,26,333,541,554,445,48,22,54,36,98,11,25,
158,256,332,428,22,56,487,54,26,333,541,554,445,48,22,54,36,98,11,25};
Aurora::Matrix v3(std::shared_ptr<double>(v3D,std::default_delete<double[]>()),std::vector<int>{5,4,3});
Aurora::Matrix v3(std::shared_ptr<float>(v3D,std::default_delete<float[]>()),std::vector<int>{5,4,3});
Aurora::Matrix result = interpn(y,x,v2,x1,y1,Aurora::InterpnMethod::Spline);
EXPECT_DOUBLE_AE(result.getData()[0],399.2742);
EXPECT_DOUBLE_AE(result.getData()[1],387.7651);
EXPECT_DOUBLE_AE(result.getData()[2],135.6138);
EXPECT_FLOAT_AE(result.getData()[0],399.2742);
EXPECT_FLOAT_AE(result.getData()[1],387.7651);
EXPECT_FLOAT_AE(result.getData()[2],135.6138);
result = interpn(y,x,v2,x1,y1,Aurora::InterpnMethod::Linear);
EXPECT_DOUBLE_AE(result.getData()[0],352.1727);
EXPECT_DOUBLE_AE(result.getData()[1],269.8596);
EXPECT_DOUBLE_AE(result.getData()[2],94.7908);
EXPECT_FLOAT_AE(result.getData()[0],352.1727);
EXPECT_FLOAT_AE(result.getData()[1],269.8596);
EXPECT_FLOAT_AE(result.getData()[2],94.7908);
result = interpn(y,x,z,v3,x1,y1,z1,Aurora::InterpnMethod::Spline);
EXPECT_DOUBLE_AE(result.getData()[0],399.2742);
EXPECT_DOUBLE_AE(result.getData()[1],387.7651);
EXPECT_DOUBLE_AE(result.getData()[2],135.6138);
EXPECT_FLOAT_AE(result.getData()[0],399.2742);
EXPECT_FLOAT_AE(result.getData()[1],387.7651);
EXPECT_FLOAT_AE(result.getData()[2],135.6138);
result = interpn(y,x,z,v3,x1,y1,z1,Aurora::InterpnMethod::Linear);
EXPECT_DOUBLE_AE(result.getData()[0],352.1727);
EXPECT_DOUBLE_AE(result.getData()[1],269.8596);
EXPECT_DOUBLE_AE(result.getData()[2],94.7908);
EXPECT_FLOAT_AE(result.getData()[0],352.1727);
EXPECT_FLOAT_AE(result.getData()[1],269.8596);
EXPECT_FLOAT_AE(result.getData()[2],94.7908);
}
TEST_F(Function3D_Test, zerosAndones){
@@ -118,25 +118,25 @@ TEST_F(Function3D_Test, zerosAndones){
Aurora::Matrix onesM = Aurora::ones( 9, 9,9);
EXPECT_EQ(729,onesM.getDataSize());
for (int i = 0; i < onesM.getDataSize(); ++i) {
EXPECT_DOUBLE_EQ(1.0,onesM.getData()[i])<<" error at index："<<i;
EXPECT_FLOAT_EQ(1.0,onesM.getData()[i])<<" error at index："<<i;
}
onesM = Aurora::ones( 9, 9);
EXPECT_EQ(81,onesM.getDataSize());
for (int i = 0; i < onesM.getDataSize(); ++i) {
EXPECT_DOUBLE_EQ(1.0,onesM.getData()[i])<<" error at index："<<i;
EXPECT_FLOAT_EQ(1.0,onesM.getData()[i])<<" error at index："<<i;
}
onesM = Aurora::ones( 9, 1);
EXPECT_EQ(9,onesM.getDataSize());
for (int i = 0; i < onesM.getDataSize(); ++i) {
EXPECT_DOUBLE_EQ(1.0,onesM.getData()[i])<<" error at index："<<i;
EXPECT_FLOAT_EQ(1.0,onesM.getData()[i])<<" error at index："<<i;
}
onesM = Aurora::ones( 9);
EXPECT_EQ(81,onesM.getDataSize());
for (int i = 0; i < onesM.getDataSize(); ++i) {
EXPECT_DOUBLE_EQ(1.0,onesM.getData()[i])<<" error at index："<<i;
EXPECT_FLOAT_EQ(1.0,onesM.getData()[i])<<" error at index："<<i;
}
}