test batch

2020-04-23 00:54:03 +02:00
parent 8caff5f598
commit 488887992c
2 changed files with 29 additions and 66 deletions
@@ -51,7 +51,7 @@ int main(int argc, char *argv[]) {
                dnnType *out_d = (dnnType*) netRT.buffersRT[o];
                dnnType *out0_d = out_d;
                dnnType *outI_d = out_d + netRT.buffersDIM[o].tot()*b;
-                ret_tensorrt |= checkResult(netRT.buffersDIM[o].tot(), outI_d, out0_d);
+                ret_tensorrt |= checkResult(netRT.buffersDIM[o].tot(), outI_d, out0_d) == 0 ? 0 : ERROR_TENSORRT;
            }
        }
    }