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

Merge branch 'dtof' of http://192.168.1.9:3000/Bug/Aurora into dtof

Browse Source
This commit is contained in:
kradchen
2023-12-08 17:32:03 +08:00
parent 4eeb6c4988 0082800112
commit b0a2ac39fb
3 changed files with 126 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
test/Function2D_Cuda_Test.cpp
View File

@@ -793,4 +793,24 @@ TEST_F(Function2D_Cuda_Test, inv) {
{
EXPECT_FLOAT_AE(result1[i], result2[i]);
}
}
TEST_F(Function2D_Cuda_Test, dot) {
auto matrixHost1 = Aurora::Matrix::fromRawData(new float[200000], 1000,200);
auto matrixHost2 = Aurora::Matrix::fromRawData(new float[200000], 1000,200);
for(unsigned int i=0; i<200000;++i)
{
matrixHost1[i] = i;
matrixHost2[i] = i + 1;
}
auto matrixDevice1 = matrixHost1.toDeviceMatrix();
Aurora::CudaMatrix matrixDevice2 = matrixHost2.toDeviceMatrix();
auto result1 = Aurora::dot(matrixHost1, matrixHost2);
auto result2 = Aurora::dot(matrixDevice1, matrixDevice2).toHostMatrix();
std::cout<< result1.getDataSize();
ASSERT_FLOAT_EQ(result1.getDataSize(), result2.getDataSize());
for (size_t i = 0; i < result1.getDataSize(); i++)
{
EXPECT_FLOAT_AE(result1[i], result2[i]);
}
}