add warm-up
This commit is contained in:
parent
a01253f169
commit
f79af947fe
|
@ -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);
|
||||
|
||||
|
|
Loading…
Reference in New Issue