is_kernel_expression.hpp

Go to the documentation of this file.

#ifndef STAN_MATH_OPENCL_KERNEL_GENERATOR_IS_KERNEL_EXPRESSION_HPP
#define STAN_MATH_OPENCL_KERNEL_GENERATOR_IS_KERNEL_EXPRESSION_HPP
 
#include <stan/math/prim/meta/bool_constant.hpp>
#include <stan/math/prim/meta/conjunction.hpp>
#include <stan/math/prim/meta/disjunction.hpp>
#include <stan/math/prim/meta/is_matrix_cl.hpp>
#include <stan/math/prim/meta/is_var.hpp>
#include <stan/math/prim/meta/require_helpers.hpp>
#include <type_traits>
 
namespace stan {
 
class operation_cl_base {};
class operation_cl_lhs_base {};
 
template <typename T, typename = void>
struct is_kernel_expression_and_not_scalar
    : bool_constant<std::is_base_of<operation_cl_base,
                                    std::remove_reference_t<T>>::value> {};
template <typename T>
struct is_kernel_expression_and_not_scalar<T, require_matrix_cl_t<T>>
    : std::true_type {};
 
template <typename T>
struct is_kernel_expression
    : bool_constant<is_kernel_expression_and_not_scalar<T>::value
                    || std::is_arithmetic<std::remove_reference_t<T>>::value> {
};
 
template <typename... Types>
using require_all_kernel_expressions_and_none_scalar_t
    = require_all_t<is_kernel_expression_and_not_scalar<Types>...>;
 
template <typename... Types>
using require_all_kernel_expressions_t
    = require_all_t<is_kernel_expression<Types>...>;
 
template <typename T, typename = void>
struct is_kernel_expression_lhs
    : bool_constant<std::is_base_of<operation_cl_lhs_base,
                                    std::remove_reference_t<T>>::value> {};
template <typename T>
struct is_kernel_expression_lhs<T, require_matrix_cl_t<T>> : std::true_type {};
 
template <typename T>
struct is_rev_kernel_expression
    : math::conjunction<is_var<T>, is_kernel_expression<value_type_t<T>>> {};
 
template <typename T>
struct is_prim_or_rev_kernel_expression
    : math::disjunction<is_kernel_expression<T>, is_rev_kernel_expression<T>> {
};
 
template <typename T>
struct is_nonscalar_prim_or_rev_kernel_expression
    : math::disjunction<
          is_kernel_expression_and_not_scalar<T>,
          math::conjunction<is_var<T>, is_kernel_expression_and_not_scalar<
                                           value_type_t<T>>>> {};
 
template <typename T>
using require_kernel_expression_lhs_t
    = require_t<is_kernel_expression_lhs<std::decay_t<T>>>;
template <typename T>
using require_rev_kernel_expression_t
    = require_t<is_rev_kernel_expression<std::decay_t<T>>>;
 
template <typename T>
using require_not_rev_kernel_expression_t
    = require_not_t<is_rev_kernel_expression<std::decay_t<T>>>;
template <typename... Types>
using require_all_prim_or_rev_kernel_expression_t
    = require_all_t<is_prim_or_rev_kernel_expression<std::decay_t<Types>>...>;
template <typename T>
using require_nonscalar_prim_or_rev_kernel_expression_t
    = require_t<is_nonscalar_prim_or_rev_kernel_expression<std::decay_t<T>>>;
 
template <typename T>
using require_not_nonscalar_prim_or_rev_kernel_expression_t = require_not_t<
    is_nonscalar_prim_or_rev_kernel_expression<std::decay_t<T>>>;
 
template <typename... Types>
using require_all_nonscalar_prim_or_rev_kernel_expression_t = require_all_t<
    is_nonscalar_prim_or_rev_kernel_expression<std::decay_t<Types>>...>;
 
template <typename... Types>
using require_any_nonscalar_prim_or_rev_kernel_expression_t = require_any_t<
    is_nonscalar_prim_or_rev_kernel_expression<std::decay_t<Types>>...>;
 
template <typename... Types>
using require_all_not_nonscalar_prim_or_rev_kernel_expression_t
    = require_all_not_t<
        is_nonscalar_prim_or_rev_kernel_expression<std::decay_t<Types>>...>;
}  // namespace stan
 
#endif