as_operation_cl.hpp

Go to the documentation of this file.

#ifndef STAN_MATH_OPENCL_KERNEL_GENERATOR_AS_OPERATION_CL_HPP
#define STAN_MATH_OPENCL_KERNEL_GENERATOR_AS_OPERATION_CL_HPP
#ifdef STAN_OPENCL
 
#include <stan/math/opencl/kernel_generator/assignment_ops.hpp>
#include <stan/math/opencl/kernel_generator/operation_cl.hpp>
#include <stan/math/opencl/kernel_generator/load.hpp>
#include <stan/math/opencl/kernel_generator/scalar.hpp>
#include <stan/math/opencl/matrix_cl.hpp>
#include <stan/math/prim/meta.hpp>
#include <type_traits>
 
namespace stan {
namespace math {
 
template <assign_op_cl AssignOp = assign_op_cl::equals, typename T_operation,
          typename = std::enable_if_t<std::is_base_of<
              operation_cl_base, std::remove_reference_t<T_operation>>::value>>
inline T_operation&& as_operation_cl(T_operation&& a) {
  return std::forward<T_operation>(a);
}
 
template <assign_op_cl AssignOp = assign_op_cl::equals, typename T_scalar,
          typename = require_arithmetic_t<T_scalar>,
          require_not_same_t<T_scalar, bool>* = nullptr>
inline scalar_<T_scalar> as_operation_cl(const T_scalar a) {
  return scalar_<T_scalar>(a);
}
 
template <assign_op_cl AssignOp = assign_op_cl::equals>
inline scalar_<char> as_operation_cl(const bool a) {
  return scalar_<char>(a);
}
 
template <assign_op_cl AssignOp = assign_op_cl::equals, typename T_matrix_cl,
          typename = require_any_t<is_matrix_cl<T_matrix_cl>,
                                   is_arena_matrix_cl<T_matrix_cl>>>
inline load_<T_matrix_cl, AssignOp> as_operation_cl(T_matrix_cl&& a) {
  return load_<T_matrix_cl, AssignOp>(std::forward<T_matrix_cl>(a));
}
 
template <typename T, assign_op_cl AssignOp = assign_op_cl::equals>
using as_operation_cl_t
    = std::conditional_t<std::is_lvalue_reference<T>::value,
                         decltype(as_operation_cl<AssignOp>(std::declval<T>())),
                         std::remove_reference_t<decltype(
                             as_operation_cl<AssignOp>(std::declval<T>()))>>;
 
}  // namespace math
}  // namespace stan
 
#endif
#endif