Stan Math Library
5.0.0
Automatic Differentiation
|
__kernel void stan::math::opencl_kernels::cholesky_decompose | ( | __global double * | A, |
int | rows | ||
) |
Calculates the Cholesky Decomposition of a matrix on an OpenCL.
This kernel is run with threads organized in one dimension and in a single thread block. The kernel is best suited for small input matrices as it only utilizes a single streaming multiprocessor. The kernels is used as a part of a blocked cholesky decompose.
[in,out] | A | The input matrix and the result of the cholesky decomposition |
rows | The number of rows for A and B. |
const char*
held in cholesky_decompose_kernel_code.
Used in math/opencl/cholesky_decompose.hpp. This kernel uses the helper macros available in helpers.cl. Definition at line 33 of file cholesky_decompose.hpp.