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

Add cuda convertfp16tofloat

Browse Source
This commit is contained in:
kradchen
2023-12-18 13:48:35 +08:00
parent b09257298c
commit fd7c71f7e9
4 changed files with 59 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
test/Function1D_Cuda_Test.cpp
View File

@@ -1,5 +1,6 @@
#include <gtest/gtest.h>
#include "MatlabReader.h"
#include "CudaMatrix.h"
#include "Matrix.h"
#include "TestUtility.h"
@@ -1149,3 +1150,16 @@ TEST_F(Function1D_Cuda_Test, uniqueByRows) {
EXPECT_FLOAT_AE(indexResult1[i], indexResult2[i]);
}
}
TEST_F(Function1D_Cuda_Test, convertfp16tofloat) {
MatlabReader m("/home/krad/TestData/convertReal.mat");
size_t count = 0;
auto input = m.readint16("input",count);
auto resultM = Aurora::convertfp16tofloat(input.get(),count,1);
auto resultC = Aurora::convertfp16tofloatCuda(input.get(),count,1);
for (size_t i = 0; i<100; i++) {
EXPECT_FLOAT_EQ(resultC.getValue(i), resultM[i])<<",index:"<<i;
}
}

3
test/Function1D_Test.cpp
View File

@@ -571,9 +571,6 @@ TEST_F(Function1D_Test, convertfp16tofloat) {
size_t count = 0;
auto input = m.readint16("input",count);
for(int i = 0; i<1000; i++){
auto resultM = Aurora::convertfp16tofloat(input.get(),count,1);
}
auto resultM = Aurora::convertfp16tofloat(input.get(),count,1);
auto result = resultM.getData();
auto output = m.read("output");