Add cuda convertfp16tofloat

2023-12-18 13:48:35 +08:00
parent b09257298c
commit fd7c71f7e9
4 changed files with 59 additions and 3 deletions
--- a/test/Function1D_Test.cpp
+++ b/test/Function1D_Test.cpp
@@ -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");