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

Add Cuda matrix size function & unittest

Browse Source
This commit is contained in:
kradchen
2023-12-15 10:54:29 +08:00
parent d070edfef7
commit 6dfea91ed0
3 changed files with 176 additions and 29 deletions
Download Patch File Download Diff File Expand all files Collapse all files

102
test/Function3D_Test.cpp
View File

@@ -4,6 +4,8 @@
#include "TestUtility.h"
#include "Matrix.h"
#include "CudaMatrix.h"
#include "Function.h"
#include "Function1D.h"
#include "Function2D.h"
@@ -93,50 +95,98 @@ TEST_F(Function3D_Test, interpn) {
EXPECT_FLOAT_AE(result.getData()[2],94.7908);
}
TEST_F(Function3D_Test, zerosAndones){
Aurora::Matrix zerosM = Aurora::zeros( 3, 4,5);
EXPECT_EQ(60,zerosM.getDataSize());
TEST_F(Function3D_Test, zerosAndonesAndsizesMKL) {
// mklversion
Aurora::Matrix zerosM = Aurora::zeros(3, 4, 5);
EXPECT_EQ(60, zerosM.getDataSize());
auto size = Aurora::size(zerosM);
EXPECT_EQ(3,size[0]);
EXPECT_EQ(4,size[1]);
EXPECT_EQ(5,size[2]);
EXPECT_EQ(3,Aurora:: size(zerosM,1));
EXPECT_EQ(4,Aurora:: size(zerosM,2));
EXPECT_EQ(5,Aurora:: size(zerosM,3));
for (int i = 0; i < zerosM.getDataSize(); ++i) {
EXPECT_EQ(0,zerosM.getData()[i])<<" error at index："<<i;
EXPECT_EQ(0, zerosM.getData()[i]) << " error at index:" << i;
}
zerosM = Aurora::zeros( 3, 4);
EXPECT_EQ(12,zerosM.getDataSize());
zerosM = Aurora::zeros(3, 4);
EXPECT_EQ(12, zerosM.getDataSize());
for (int i = 0; i < zerosM.getDataSize(); ++i) {
EXPECT_EQ(0,zerosM.getData()[i])<<" error at index："<<i;
EXPECT_EQ(0, zerosM.getData()[i]) << " error at index:" << i;
}
zerosM = Aurora::zeros( 3, 1);
EXPECT_EQ(3,zerosM.getDataSize());
zerosM = Aurora::zeros(3, 1);
EXPECT_EQ(3, zerosM.getDataSize());
for (int i = 0; i < zerosM.getDataSize(); ++i) {
EXPECT_EQ(0,zerosM.getData()[i])<<" error at index："<<i;
EXPECT_EQ(0, zerosM.getData()[i]) << " error at index:" << i;
}
zerosM = Aurora::zeros( 20);
EXPECT_EQ(400,zerosM.getDataSize());
zerosM = Aurora::zeros(20);
EXPECT_EQ(400, zerosM.getDataSize());
for (int i = 0; i < zerosM.getDataSize(); ++i) {
EXPECT_EQ(0,zerosM.getData()[i])<<" error at index："<<i;
EXPECT_EQ(0, zerosM.getData()[i]) << " error at index:" << i;
}
Aurora::Matrix onesM = Aurora::ones( 9, 9,9);
EXPECT_EQ(729,onesM.getDataSize());
Aurora::Matrix onesM = Aurora::ones(9, 9, 9);
EXPECT_EQ(729, onesM.getDataSize());
for (int i = 0; i < onesM.getDataSize(); ++i) {
EXPECT_FLOAT_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());
onesM = Aurora::ones(9, 9);
EXPECT_EQ(81, onesM.getDataSize());
for (int i = 0; i < onesM.getDataSize(); ++i) {
EXPECT_FLOAT_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());
onesM = Aurora::ones(9, 1);
EXPECT_EQ(9, onesM.getDataSize());
for (int i = 0; i < onesM.getDataSize(); ++i) {
EXPECT_FLOAT_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());
onesM = Aurora::ones(9);
EXPECT_EQ(81, onesM.getDataSize());
for (int i = 0; i < onesM.getDataSize(); ++i) {
EXPECT_FLOAT_EQ(1.0,onesM.getData()[i])<<" error at index："<<i;
EXPECT_FLOAT_EQ(1.0, onesM.getData()[i]) << " error at index:" << i;
}
}
TEST_F(Function3D_Test, zerosAndonesAndsizeCuda) {
Aurora::CudaMatrix zerosM = Aurora::zerosCuda(3, 4, 5);
auto size = Aurora::size(zerosM);
EXPECT_EQ(3,size.getValue(0));
EXPECT_EQ(4,size.getValue(1));
EXPECT_EQ(5,size.getValue(2));
EXPECT_EQ(3,Aurora:: size(zerosM,1));
EXPECT_EQ(4,Aurora:: size(zerosM,2));
EXPECT_EQ(5,Aurora:: size(zerosM,3));
EXPECT_EQ(60, zerosM.getDataSize());
for (int i = 0; i < zerosM.getDataSize(); ++i) {
EXPECT_EQ(0, zerosM.getValue(i)) << " error at index:" << i;
}
zerosM = Aurora::zerosCuda(3, 4);
EXPECT_EQ(12, zerosM.getDataSize());
for (int i = 0; i < zerosM.getDataSize(); ++i) {
EXPECT_EQ(0, zerosM.getValue(i)) << " error at index:" << i;
}
zerosM = Aurora::zerosCuda(20);
EXPECT_EQ(400, zerosM.getDataSize());
for (int i = 0; i < zerosM.getDataSize(); ++i) {
EXPECT_EQ(0, zerosM.getValue(i)) << " error at index:" << i;
}
auto onesM = Aurora::onesCuda(9, 9, 9);
EXPECT_EQ(729, onesM.getDataSize());
for (int i = 0; i < onesM.getDataSize(); ++i) {
ASSERT_FLOAT_EQ(1.0, onesM.getValue(i)) << " error at index:" << i;
}
onesM = Aurora::onesCuda(9, 3);
EXPECT_EQ(27, onesM.getDataSize());
for (int i = 0; i < onesM.getDataSize(); ++i) {
EXPECT_FLOAT_EQ(1.0, onesM.getValue(i)) << " error at index:" << i;
}
onesM = Aurora::onesCuda(9);
EXPECT_EQ(81, onesM.getDataSize());
for (int i = 0; i < onesM.getDataSize(); ++i) {
EXPECT_FLOAT_EQ(1.0, onesM.getValue(i)) << " error at index:" << i;
}
}