fill_view1D.cpp

Go to the documentation of this file.

#include "../../../include/common/monolish_dense.hpp"
#include "../../../include/common/monolish_logger.hpp"
#include "../../../include/common/monolish_matrix.hpp"
#include "../../../include/monolish_vml.hpp"
#include "../../internal/monolish_internal.hpp"
 
namespace monolish {
 
template <typename T, typename U> void view1D<T, U>::fill(U value) {
  Logger &logger = Logger::get_instance();
  logger.util_in(monolish_func);
  U *val = data();
  if (get_device_mem_stat() == true) {
#if MONOLISH_USE_GPU
#pragma omp target teams distribute parallel for
    for (size_t i = first; i < last; i++) {
      val[i] = value;
    }
#else
    throw std::runtime_error(
        "error USE_GPU is false, but get_device_mem_stat() == true");
#endif
  } else {
#pragma omp parallel for
    for (size_t i = first; i < last; i++) {
      val[i] = value;
    }
  }
  logger.util_out();
}
template void view1D<vector<double>, double>::fill(double value);
template void view1D<matrix::Dense<double>, double>::fill(double value);
template void
view1D<matrix::LinearOperator<double>, double>::fill(double value);
template void view1D<vector<float>, float>::fill(float value);
template void view1D<matrix::Dense<float>, float>::fill(float value);
template void view1D<matrix::LinearOperator<float>, float>::fill(float value);
 
} // namespace monolish