Stan Math Library  2.20.0
reverse mode automatic differentiation
Namespaces | Functions | Variables
add.hpp File Reference
#include <stan/math/opencl/kernel_cl.hpp>
#include <stan/math/opencl/buffer_types.hpp>

Go to the source code of this file.

Namespaces

 stan
 
 stan::math
 Matrices and templated mathematical functions.
 
 stan::math::opencl_kernels
 

Functions

__kernel void stan::math::opencl_kernels::add (__global double *C, __global double *A, __global double *B, unsigned int rows, unsigned int cols)
 Matrix addition on the OpenCL device. More...
 
__kernel void stan::math::opencl_kernels::add_batch (__global double *B, __global double *A, unsigned int rows, unsigned int cols, unsigned int batch_size)
 Sums a batch of matrices. More...
 

Variables

const kernel_cl< out_buffer, in_buffer, in_buffer, int, int > stan::math::opencl_kernels::add ("add", {indexing_helpers, add_kernel_code})
 See the docs for add() . More...
 
const kernel_cl< out_buffer, in_buffer, int, int, int > stan::math::opencl_kernels::add_batch ("add_batch", {indexing_helpers, add_batch_kernel_code})
 See the docs for add_batch() . More...
 

     [ Stan Home Page ] © 2011–2018, Stan Development Team.