Stan Math Library
5.0.0
Automatic Differentiation
|
#include <stan/math/opencl/kernel_cl.hpp>
#include <stan/math/opencl/buffer_types.hpp>
#include <stan/math/opencl/matrix_cl_view.hpp>
#include <string>
Go to the source code of this file.
Classes | |
struct | stan::math::opencl_kernels::cumulative_sum< Scalar, typename > |
struct containing cumulative_sum kernels, grouped by scalar type. More... | |
struct | stan::math::opencl_kernels::cumulative_sum< double, T > |
struct | stan::math::opencl_kernels::cumulative_sum< int, T > |
Namespaces | |
namespace | stan |
The lgamma implementation in stan-math is based on either the reentrant safe lgamma_r implementation from C or the boost::math::lgamma implementation. | |
namespace | stan::math |
Matrices and templated mathematical functions. | |
namespace | stan::math::opencl_kernels |
Functions | |
__kernel void | stan::math::opencl_kernels::cumulative_sum1 (__global SCAL *out_wgs, __global SCAL *out_threads, __global SCAL *in, int size) |
First kernel of the cumulative sum implementation. | |
__kernel void | stan::math::opencl_kernels::cumulative_sum2 (__global SCAL *data, int size) |
Second kernel of the cumulative sum implementation. | |
__kernel void | stan::math::opencl_kernels::cumulative_sum3 (__global SCAL *out, __global SCAL *in_data, __global SCAL *in_threads, __global SCAL *in_wgs, int size) |
Third kernel of the cumulative sum implementation. | |