Snpemodel: Clean up after #23772 (#23780)

* #clean up snpemodel after #23772

* Leave benchmark remove loop
pull/23782/head
redacid95 4 years ago committed by GitHub
parent 9535dc606d
commit 1c9bf108af
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
  1. 13
      selfdrive/modeld/runners/snpemodel.cc

@ -159,14 +159,11 @@ void SNPEModel::execute(float *net_input_buf, int buf_size) {
float *outputs_golden = (float *)malloc(output_size*sizeof(float));
memcpy(outputs_golden, output, output_size*sizeof(float));
memset(output, 0, output_size*sizeof(float));
for (int i = 0; i < 5; i++) {
memset(recurrent, 0, recurrent_size*sizeof(float));
uint64_t start_time = nanos_since_boot();
thneed->execute(inputs, output);
uint64_t elapsed_time = nanos_since_boot() - start_time;
printf("ran model in %.2f ms\n", float(elapsed_time)/1e6);
}
memset(recurrent, 0, recurrent_size*sizeof(float));
uint64_t start_time = nanos_since_boot();
thneed->execute(inputs, output);
uint64_t elapsed_time = nanos_since_boot() - start_time;
printf("ran model in %.2f ms\n", float(elapsed_time)/1e6);
if (memcmp(output, outputs_golden, output_size*sizeof(float)) == 0) {
printf("thneed selftest passed\n");

Loading…
Cancel
Save