ops_mpi.hpp

#pragma once

#include <vector>

#include "core/task/include/task.hpp"

namespace nesterov_a_test_task_mpi {

void MultiplyRowMajor(const std::vector<int> &in, std::vector<int> &out, int rc_size);
void MultiplyColumnMajor(const std::vector<int> &in, std::vector<int> &out, int rc_size);

class TestTaskMPI : public ppc::core::Task {
 public:
  explicit TestTaskMPI(const std::vector<int> &in) : input_(in) {}
  bool ValidationImpl() override;
  bool PreProcessingImpl() override;
  bool RunImpl() override;
  bool PostProcessingImpl() override;
  std::vector<int> Get();

 private:
  std::vector<int> input_, output_;
  int rc_size_{};

  void MultiplyMatrixBasedOnRank();
};

}  // namespace nesterov_a_test_task_mpi