add warm-up

This commit is contained in:
bhko 2023-02-06 06:21:16 +00:00
parent a01253f169
commit f79af947fe
1 changed files with 12 additions and 7 deletions

View File

@ -34,19 +34,24 @@ int main(int argc, char **argv) {
fflush(stdout);
// warm_up
print("warm_up.");
for(int i = 0 ; i < WARM_UP ; ++i)
printf("\nWarmimg up.");
for(int i = 0 ; i < WARM_UP ; ++i){
uNet(input, output, N);
print(".");
printf(".");
}
cudaDeviceSynchronize();
print("\n");
printf("\n");
// run uNet and measure time
printf("\nProcess.");
double uNet_st = get_time();
for(int j = 0 ; j < MEASURE ; ++j)
for(int j = 0 ; j < MEASURE ; ++j){
uNet(input, output, N);
printf(".");
}
cudaDeviceSynchronize();
double uNet_en = get_time();
printf("\n");
double elapsed_time = uNet_en - uNet_st;
printf("Done! (%lf img/sec)\n", N/elapsed_time/MEASURE);