diff --git a/cs677/final/mpi-fractals.cc b/cs677/final/mpi-fractals.cc index d8d4bd3..e62af76 100644 --- a/cs677/final/mpi-fractals.cc +++ b/cs677/final/mpi-fractals.cc @@ -52,6 +52,7 @@ int main(int argc, char * argv[]) Computation * computation = NULL; int fractal_type = 0; bool display_times = false; + bool one_run = false; SDL_Surface * screen; Uint32 * pixels; @@ -83,6 +84,10 @@ int main(int argc, char * argv[]) { task_size = atoi(strlen(argv[i]) > 2 ? argv[i] + 2 : argv[++i]); } + else if (!strcmp(argv[i], "--one-run")) + { + one_run = true; + } } getSizes(&my_rank, &world_size, &nprocs); if (my_rank == 0) @@ -135,6 +140,8 @@ int main(int argc, char * argv[]) gettimeofday(&before, NULL); draw(my_rank, world_size, nprocs, width, height, pixels, taskVals, computation); + if (one_run) + going = false; gettimeofday(&after, NULL); if (display_times) {