![]() |
Stan Math Library
2.20.0
reverse mode automatic differentiation
|
#include <stan/math/opencl/opencl_context.hpp>
#include <stan/math/opencl/kernel_cl.hpp>
#include <stan/math/opencl/matrix_cl.hpp>
#include <stan/math/opencl/kernels/copy.hpp>
#include <stan/math/opencl/kernels/pack.hpp>
#include <stan/math/opencl/kernels/unpack.hpp>
#include <stan/math/opencl/buffer_types.hpp>
#include <stan/math/opencl/err/check_opencl.hpp>
#include <stan/math/prim/mat/fun/Eigen.hpp>
#include <stan/math/prim/scal/err/check_size_match.hpp>
#include <stan/math/prim/arr/fun/vec_concat.hpp>
#include <CL/cl.hpp>
#include <iostream>
#include <vector>
#include <algorithm>
#include <type_traits>
Go to the source code of this file.
Namespaces | |
stan | |
stan::math | |
Matrices and templated mathematical functions. | |
Functions | |
template<int R, int C> | |
matrix_cl | stan::math::to_matrix_cl (const Eigen::Matrix< double, R, C > &src) |
Copies the source Eigen matrix to the destination matrix that is stored on the OpenCL device. More... | |
Eigen::Matrix< double, Eigen::Dynamic, Eigen::Dynamic > | stan::math::from_matrix_cl (const matrix_cl &src) |
Copies the source matrix that is stored on the OpenCL device to the destination Eigen matrix. More... | |
template<TriangularViewCL triangular_view> | |
std::vector< double > | stan::math::packed_copy (const matrix_cl &src) |
Packs the flat triagnular matrix on the OpenCL device and copies it to the std::vector. More... | |
template<TriangularViewCL triangular_view> | |
matrix_cl | stan::math::packed_copy (const std::vector< double > &src, int rows) |
Copies the packed triangular matrix from the source std::vector to an OpenCL buffer and unpacks it to a flat matrix on the OpenCL device. More... | |
matrix_cl | stan::math::copy_cl (const matrix_cl &src) |
Copies the source matrix to the destination matrix. More... | |
template<typename T , std::enable_if_t< std::is_arithmetic< T >::value, int > = 0> | |
matrix_cl | stan::math::to_matrix_cl (const T &src) |
Copy an arithmetic type to the device. More... | |