ub_constrain.hpp

Go to the documentation of this file.

#ifndef STAN_MATH_OPENCL_PRIM_UB_CONSTRAIN_HPP
#define STAN_MATH_OPENCL_PRIM_UB_CONSTRAIN_HPP
#ifdef STAN_OPENCL
 
#include <stan/math/opencl/prim/sum.hpp>
#include <stan/math/opencl/matrix_cl.hpp>
#include <stan/math/opencl/kernel_generator.hpp>
#include <stan/math/prim/fun/constants.hpp>
 
namespace stan {
namespace math {
 
template <typename T, typename U,
          require_all_kernel_expressions_and_none_scalar_t<T>* = nullptr,
          require_all_kernel_expressions_t<U>* = nullptr>
inline auto ub_constrain(T&& x, U&& ub) {
  return make_holder_cl(
      [](auto& x_, auto& ub_) {
        return select(ub_ == INFTY, x_, ub_ - exp(x_));
      },
      std::forward<T>(x), std::forward<U>(ub));
}
 
template <typename T, typename U,
          require_all_kernel_expressions_and_none_scalar_t<T>* = nullptr,
          require_all_kernel_expressions_t<U>* = nullptr>
inline auto ub_constrain(const T& x, const U& ub, return_type_t<T, U>& lp) {
  matrix_cl<double> lp_inc;
  matrix_cl<double> res;
  auto ub_inf = ub == INFTY;
  auto lp_inc_expr = sum_2d(select(ub_inf, 0.0, x));
  auto res_expr = select(ub_inf, x, ub - exp(x));
  results(lp_inc, res) = expressions(lp_inc_expr, res_expr);
  lp += sum(from_matrix_cl(lp_inc));
  return res;
}
 
}  // namespace math
}  // namespace stan
#endif
#endif