mirror of
https://github.com/firestar5683/StarPilot.git
synced 2026-06-29 18:42:07 +08:00
* #clean up snpemodel after #23772 * Leave benchmark remove loop old-commit-hash: 1c9bf108af71546acbea291c048f3c783ce994d2
This commit is contained in:
@@ -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");
|
||||
|
||||
Reference in New Issue
Block a user