rep_matrix.hpp

Go to the documentation of this file.

#ifndef STAN_MATH_OPENCL_PRIM_REP_MATRIX_HPP
#define STAN_MATH_OPENCL_PRIM_REP_MATRIX_HPP
#ifdef STAN_OPENCL
 
#include <stan/math/prim/meta.hpp>
#include <stan/math/prim/err.hpp>
#include <stan/math/opencl/matrix_cl.hpp>
#include <stan/math/opencl/kernel_generator.hpp>
#include <stan/math/opencl/err.hpp>
 
namespace stan {
namespace math {
template <typename T, require_matrix_cl_t<T>* = nullptr>
inline auto rep_matrix(const value_type_t<T>& x, int n, int m) {
  check_nonnegative("rep_matrix (OpenCL)", "rows", n);
  check_nonnegative("rep_matrix (OpenCL)", "cols", m);
  return constant(x, n, m);
}
 
template <typename T, require_arithmetic_t<T>* = nullptr>
inline auto rep_matrix(const matrix_cl<T>& x, int m) {
  check_nonnegative("rep_matrix (OpenCL)", "rows/columns", m);
  check_mat_not_size_one("rep_matrix (OpenCL)", "x", x);
  check_vector("rep_matrix (OpenCL)", "x", x);
  const int N = x.rows() == 1 ? m : x.rows();
  const int M = x.cols() == 1 ? m : x.cols();
  return indexing(x, row_index(N, M) % x.rows(), col_index() % x.cols());
}
 
}  // namespace math
}  // namespace stan
 
#endif
#endif