math/opencl_2prim_2identity__matrix_8hpp_source.html

#ifndef STAN_MATH_OPENCL_PRIM_IDENTITY_MATRIX_HPP

#define STAN_MATH_OPENCL_PRIM_IDENTITY_MATRIX_HPP

#ifdef STAN_OPENCL


#include <stan/math/opencl/matrix_cl.hpp>

#include <stan/math/opencl/kernel_generator.hpp>


namespace stan {

namespace math {


template <typename T_x, require_matrix_cl_t<T_x>* = nullptr>

inline auto identity_matrix(int K) {

  using T_val = value_type_t<T_x>;

  check_nonnegative("identity_matrix(OpenCL)", "size", K);

  return select(row_index(K, K) == col_index(), T_val(1), T_val(0));

}

}  // namespace math

}  // namespace stan

#endif

#endif

stan::math::col_index
Represents operation that determines column index.
Definition index.hpp:80

stan::math::row_index
Represents operation that determines row index.
Definition index.hpp:17

stan::math::select
select_< as_operation_cl_t< T_condition >, as_operation_cl_t< T_then >, as_operation_cl_t< T_else > > select(T_condition &&condition, T_then &&then, T_else &&els)
Selection operation on kernel generator expressions.
Definition select.hpp:148

stan::value_type_t
typename value_type< T >::type value_type_t
Helper function for accessing underlying type.
Definition value_type.hpp:35

kernel_generator.hpp

matrix_cl.hpp

stan::math::check_nonnegative
void check_nonnegative(const char *function, const char *name, const T_y &y)
Check if y is non-negative.
Definition check_nonnegative.hpp:24

stan::math::identity_matrix
auto identity_matrix(int K)
Return a square identity matrix.
Definition identity_matrix.hpp:19

stan
The lgamma implementation in stan-math is based on either the reentrant safe lgamma_r implementation ...
Definition unit_vector_constrain.hpp:15