all test ok

2020-06-01 16:15:29 +02:00
parent d4e0d07e09
commit c8ed6d782a
16 changed files with 63 additions and 63 deletions
@@ -42,9 +42,9 @@ int main(int argc, char *argv[]) {
        checkCuda(cudaMemcpy(input_d, input, idim.tot()*sizeof(dnnType), cudaMemcpyHostToDevice));

        tk::dnn::dataDim_t dim = idim;
-        TIMER_START
+        TKDNN_TSTART
        netRT.infer(dim, input_d);
-        TIMER_STOP
+        TKDNN_TSTOP
        total_time+= t_ns;

        // control output