gicho
diff --git a/‎CMakeLists.txt‎
Lines changed: 46 additions & 0 deletions b/‎CMakeLists.txt‎
Lines changed: 46 additions & 0 deletions
diff --git a/‎benchmark/imbalance_for/TF19.mtx.zip‎
15.2 MB b/‎benchmark/imbalance_for/TF19.mtx.zip‎
15.2 MB
diff --git a/‎benchmark/imbalance_for/main.cpp‎
Lines changed: 67 additions & 0 deletions b/‎benchmark/imbalance_for/main.cpp‎
Lines changed: 67 additions & 0 deletions
@@ -320,6 +320,7 @@ include(${TBB_ROOT}/cmake/TBBBuild.cmake)
 tbb_build(TBB_ROOT ${TBB_ROOT} CONFIG_DIR TBB_DIR MAKE_ARGS tbb_cpf=1)
 find_package(TBB REQUIRED tbb_preview)
 
+
 ## benchmark 1: wavefront computing
 message(STATUS "benchmark 1: wavefront")
 set(CMAKE_RUNTIME_OUTPUT_DIRECTORY ${TF_BENCHMARK_DIR}/wavefront)
@@ -536,6 +537,51 @@ target_link_libraries(
 )
 set_target_properties(strassen PROPERTIES COMPILE_FLAGS ${OpenMP_CXX_FLAGS})
 
+### benchmark 11: imbalance for-loop 
+#message(STATUS "benchmark 11: imbalance for-loop")
+#set(CMAKE_RUNTIME_OUTPUT_DIRECTORY ${TF_BENCHMARK_DIR}/imbalance_for)
+#add_executable(
+#  imbalance_for
+#  ${TF_BENCHMARK_DIR}/imbalance_for/main.cpp
+#  ${TF_BENCHMARK_DIR}/imbalance_for/omp.cpp
+#  ${TF_BENCHMARK_DIR}/imbalance_for/tbb.cpp
+#  ${TF_BENCHMARK_DIR}/imbalance_for/taskflow.cpp
+#  ${TF_BENCHMARK_DIR}/imbalance_for/mmio.cpp
+#)
+#target_include_directories(imbalance_for PRIVATE ${PROJECT_SOURCE_DIR}/3rd-party/CLI11)
+#target_link_libraries(
+#  imbalance_for 
+#  ${PROJECT_NAME} 
+#  Threads::Threads 
+#  ${TBB_IMPORTED_TARGETS} 
+#  ${OpenMP_CXX_LIBRARIES} 
+#  tf::default_settings
+#)
+#set_target_properties(imbalance_for PROPERTIES COMPILE_FLAGS ${OpenMP_CXX_FLAGS})
+#
+### benchmark 12: LavaMD
+#message(STATUS "benchmark 12: LavaMD")
+#set(CMAKE_RUNTIME_OUTPUT_DIRECTORY ${TF_BENCHMARK_DIR}/lavamd)
+#add_executable(
+#  lavamd
+#  ${TF_BENCHMARK_DIR}/lavamd/main.cpp
+#  ${TF_BENCHMARK_DIR}/lavamd/omp.cpp
+#  ${TF_BENCHMARK_DIR}/lavamd/tbb.cpp
+#  ${TF_BENCHMARK_DIR}/lavamd/taskflow.cpp
+#)
+#target_include_directories(lavamd PRIVATE ${PROJECT_SOURCE_DIR}/3rd-party/CLI11)
+#target_link_libraries(
+#  lavamd 
+#  ${PROJECT_NAME} 
+#  Threads::Threads 
+#  ${TBB_IMPORTED_TARGETS} 
+#  ${OpenMP_CXX_LIBRARIES} 
+#  tf::default_settings
+#)
+#set_target_properties(lavamd PROPERTIES COMPILE_FLAGS ${OpenMP_CXX_FLAGS})
+
+
+
 endif()
 
 # -----------------------------------------------------------------------------
 
@@ -0,0 +1,67 @@
+#include <CLI11.hpp>
+#include "sparse.hpp"
+
+void imbalance(
+  const std::string& model,
+  const unsigned num_threads, 
+  const unsigned num_rounds
+  ) {
+
+  double runtime {0.0};
+
+  std::string fname {"TF19.mtx"};
+  read_matrix(fname.data());
+  toCSR();
+
+  for(unsigned j=0; j<num_rounds; ++j) {
+    if(model == "tf") {
+      runtime += measure_time_taskflow(num_threads).count();
+    }
+    else if(model == "tbb") {
+      runtime += measure_time_tbb(num_threads).count();
+    }
+    else if(model == "omp") {
+      runtime += measure_time_omp(num_threads).count();
+    }
+    else assert(false);
+  }
+
+  clear();
+  
+  std::cout << std::setw(12) << runtime / num_rounds / 1e3
+            << std::endl;
+}
+
+int main(int argc, char* argv[]) {
+
+ CLI::App app{"Imbalance workload using number of non-zeros in sparse matrix"};
+
+  unsigned num_threads {1}; 
+  app.add_option("-t,--num_threads", num_threads, "number of threads (default=1)");
+
+  unsigned num_rounds {1};  
+  app.add_option("-r,--num_rounds", num_rounds, "number of rounds (default=1)");
+  
+  std::string model = "tf";
+  app.add_option("-m,--model", model, "model name tbb|omp|tf (default=tf)")
+     ->check([] (const std::string& m) {
+        if(m != "tbb" && m != "tf" && m != "omp") {
+          return "model name should be \"tbb\", \"omp\", or \"tf\"";
+        }
+        return "";
+     });
+
+  CLI11_PARSE(app, argc, argv);
+   
+  std::cout << "model=" << model << ' '
+            << "num_threads=" << num_threads << ' '
+            << "num_rounds=" << num_rounds << ' '
+            << std::endl;
+
+  if(std::ifstream input("./TF19.mtx"); !input.good()) {
+    std::cout << "Please unzip the file TF19.mtx.zip first!\n";
+    return 0;
+  }
+
+  imbalance(model, num_threads, num_rounds); 
+}