matrix_cl_view.hpp

Go to the documentation of this file.

#ifndef STAN_MATH_OPENCL_TRIANGULAR_HPP
#define STAN_MATH_OPENCL_TRIANGULAR_HPP
#ifdef STAN_OPENCL
 
#include <stan/math/opencl/stringify.hpp>
#include <stan/math/prim/fun/Eigen.hpp>
#include <type_traits>
 
namespace stan {
namespace math {
enum class matrix_cl_view { Diagonal = 0, Lower = 1, Upper = 2, Entire = 3 };
 
inline const matrix_cl_view either(const matrix_cl_view left_view,
                                   const matrix_cl_view right_view) {
  using underlying = typename std::underlying_type<matrix_cl_view>::type;
  return static_cast<matrix_cl_view>(static_cast<underlying>(left_view)
                                     | static_cast<underlying>(right_view));
}
 
inline const matrix_cl_view both(const matrix_cl_view left_view,
                                 const matrix_cl_view right_view) {
  using underlying = typename std::underlying_type<matrix_cl_view>::type;
  return static_cast<matrix_cl_view>(static_cast<underlying>(left_view)
                                     & static_cast<underlying>(right_view));
}
 
inline bool contains_nonzero(const matrix_cl_view view,
                             const matrix_cl_view part) {
  return static_cast<bool>(both(view, part));
}
 
inline const matrix_cl_view transpose(const matrix_cl_view view) {
  if (view == matrix_cl_view::Lower) {
    return matrix_cl_view::Upper;
  }
  if (view == matrix_cl_view::Upper) {
    return matrix_cl_view::Lower;
  }
  return view;
}
 
inline const matrix_cl_view invert(const matrix_cl_view view) {
  using underlying = typename std::underlying_type<matrix_cl_view>::type;
  return static_cast<matrix_cl_view>(
      static_cast<underlying>(matrix_cl_view::Entire)
      & ~static_cast<underlying>(view));
}
 
inline matrix_cl_view from_eigen_uplo_type(Eigen::UpLoType eigen_type) {
  if (eigen_type & Eigen::Lower) {
    return matrix_cl_view::Lower;
  }
  if (eigen_type & Eigen::Upper) {
    return matrix_cl_view::Upper;
  }
  return matrix_cl_view::Entire;
}
 
// \cond
static constexpr const char* view_kernel_helpers = STRINGIFY(
    // \endcond
    int either(int left_view, int right_view) { return left_view | right_view; }
 
    int both(int left_view, int right_view) { return left_view & right_view; }
 
    bool contains_nonzero(int view, int part) { return both(view, part); }
    // \cond
);
// \endcond
 
}  // namespace math
}  // namespace stan
#endif
#endif