fp16 implementation, TODO deallocate in LayerWgs

2017-08-30 14:37:25 +00:00
parent a26ef98d2d
commit 2cf8d8f6fc
11 changed files with 190 additions and 42 deletions
@@ -70,7 +70,7 @@ void printDeviceVector(int size, dnnType* vec_d, bool device)
 int checkResult(int size, dnnType *data_d, dnnType *correct_d, bool device) {

    dnnType *data_h, *correct_h;
-    const float eps = 0.0001f;
+    const float eps = 0.001f;

    if(device) {
        data_h = new dnnType[size];