Add taskwait to omp

clin99 · clin99 · commit 721ed21c2e88 · 2019-05-17T19:21:36.000-05:00
diff --git a/benchmark/mnist/omp.cpp b/benchmark/mnist/omp.cpp
@@ -180,6 +180,7 @@ void run_omp(MNIST& D, unsigned num_threads) {
          } // End of one iteration 
        } // End of one epoch 
 
+    #pragma omp taskwait
     } // End of omp single 
   } // End of omp parallel
 
