index.hpp

Go to the documentation of this file.

#ifndef STAN_MATH_OPENCL_KERNEL_GENERATOR_INDEX_HPP
#define STAN_MATH_OPENCL_KERNEL_GENERATOR_INDEX_HPP
#ifdef STAN_OPENCL
 
#include <stan/math/opencl/kernel_generator/operation_cl.hpp>
 
namespace stan {
namespace math {
 
class row_index : public operation_cl<row_index, int> {
 public:
  using Scalar = int;
  using base = operation_cl<row_index, int>;
  using base::var_name_;
  int rows_;
  int cols_;
 
  explicit row_index(int rows = base::dynamic, int cols = base::dynamic)
      : rows_(rows), cols_(cols) {}
 
  inline row_index deep_copy() const { return row_index(rows_, cols_); }
 
  inline kernel_parts generate(const std::string& row_index_name,
                               const std::string& col_index_name,
                               const bool view_handled) const {
    kernel_parts res{};
    res.body = "int " + var_name_ + " = " + row_index_name + ";\n";
    return res;
  }
 
  inline int rows() const { return rows_; }
 
  inline int cols() const { return cols_; }
 
  inline std::pair<int, int> extreme_diagonals() const {
    return {std::numeric_limits<int>::min(), std::numeric_limits<int>::max()};
  }
};
 
class col_index : public operation_cl<col_index, int> {
 public:
  using Scalar = int;
  using base = operation_cl<col_index, int>;
  using base::var_name_;
  int rows_;
  int cols_;
 
  explicit col_index(int rows = base::dynamic, int cols = base::dynamic)
      : rows_(rows), cols_(cols) {}
 
  inline col_index deep_copy() const { return col_index(rows_, cols_); }
 
  inline kernel_parts generate(const std::string& row_index_name,
                               const std::string& col_index_name,
                               const bool view_handled) const {
    kernel_parts res{};
    res.body = "int " + var_name_ + " = " + col_index_name + ";\n";
    return res;
  }
 
  inline int rows() const { return rows_; }
 
  inline int cols() const { return cols_; }
 
  inline std::pair<int, int> extreme_diagonals() const {
    return {std::numeric_limits<int>::min(), std::numeric_limits<int>::max()};
  }
};
}  // namespace math
}  // namespace stan
 
#endif
#endif