updated mnist benchmarks

tsung-wei-huang · tsung-wei-huang · commit 9c6ffbede157 · 2019-01-03T11:56:27.000-06:00
diff --git a/benchmark/mnist/omp.cpp b/benchmark/mnist/omp.cpp
@@ -5,7 +5,7 @@ void run_omp(MNIST& D, unsigned num_threads) {
 
   // Create a task flow graph
   const auto iter_num = D.images.rows()/D.batch_size;
-  const auto num_storage = num_threads * 2;
+  const auto num_storage = num_threads;
 
   // number of concurrent shuffle tasks
   const auto num_par_shf = std::min(num_storage, D.epoch);
diff --git a/benchmark/mnist/taskflow.cpp b/benchmark/mnist/taskflow.cpp
@@ -11,7 +11,7 @@ void run_taskflow(MNIST& D, unsigned num_threads) {
   std::vector<tf::Task> shuffle_tasks;
 
   // Number of parallel shuffle 
-  const auto num_storage = num_threads * 2;
+  const auto num_storage = num_threads;
   const auto num_par_shf = std::min(num_storage, D.epoch);
 
   std::vector<Eigen::MatrixXf> mats(num_par_shf, D.images);
diff --git a/benchmark/mnist/tbb.cpp b/benchmark/mnist/tbb.cpp
@@ -17,7 +17,7 @@ void run_tbb(MNIST& D, unsigned num_threads) {
   std::vector<std::unique_ptr<continue_node<continue_msg>>> shuffle_tasks;
 
   // Number of parallel shuffle 
-  const auto num_storage = num_threads * 2;
+  const auto num_storage = num_threads;
   const auto num_par_shf = std::min(num_storage, D.epoch);
 
   std::vector<Eigen::MatrixXf> mats(num_par_shf, D.images);
