main.cpp 1.34 KB
Newer Older
1 2
#include "pls/internal/scheduling/scheduler.h"
#include "pls/algorithms/for_each.h"
3

4
using namespace pls::internal::scheduling;
5

6 7
#include "benchmark_runner.h"
#include "benchmark_base/matrix.h"
8

9
using namespace comparison_benchmarks::base;
10 11

template<typename T, int SIZE>
12
class pls_matrix : public matrix::matrix<T, SIZE> {
13
 public:
14
  pls_matrix() : matrix::matrix<T, SIZE>() {}
15

16 17
  void multiply(const matrix::matrix<T, SIZE> &a, const matrix::matrix<T, SIZE> &b) override {
    pls::algorithm::for_each_range(0, SIZE, [&](int i) {
18 19 20 21 22
      this->multiply_column(i, a, b);
    });
  }
};

23
constexpr int MAX_NUM_TASKS = 32;
24
constexpr int MAX_STACK_SIZE = 4096 * 1;
25

26 27 28 29
int main(int argc, char **argv) {
  int num_threads;
  string directory;
  benchmark_runner::read_args(argc, argv, num_threads, directory);
30

31
  string test_name = to_string(num_threads) + ".csv";
32
  string full_directory = directory + "/PLS_v3/";
33
  benchmark_runner runner{full_directory, test_name};
34

35 36 37
  pls_matrix<double, matrix::MATRIX_SIZE> a;
  pls_matrix<double, matrix::MATRIX_SIZE> b;
  pls_matrix<double, matrix::MATRIX_SIZE> result;
38

39
  scheduler scheduler{(unsigned) num_threads, MAX_NUM_TASKS, MAX_STACK_SIZE};
40

41 42 43 44 45
  runner.run_iterations(matrix::NUM_ITERATIONS, [&]() {
    scheduler.perform_work([&]() {
      result.multiply(a, b);
    });
  }, matrix::WARMUP_ITERATIONS);
46 47
  runner.commit_results(true);
}