Stan Math Library
5.0.0
Automatic Differentiation
|
__kernel void stan::math::opencl_kernels::neg_rect_lower_tri_multiply | ( | __global double * | A, |
const __global double * | temp, | ||
const int | A_rows, | ||
const int | rows | ||
) |
Calculates C = -B * A where B is rectangular and A is a lower triangular.
For a full guide to the inverse lower triangular kernels see the link here.
Graphically, this kernel calculates -temp * C1
where temp is the C2 * A3 calculation from inv_lower_tri_multiply() The kernel is executed using (N, N, m) threads, where N is the size of the input matrices.
[in,out] | A | Input matrix that is being inverted. |
[in] | temp | Temporary matrix with the intermediate results. |
A_rows | Number of rows for A. | |
rows | The number of rows in a single matrix of the batch |
const char*
held in neg_rect_lower_tri_multiply_kernel_code Used in math/opencl/tri_inverse.hpp. This kernel uses the helper macros available in helpers.cl. Definition at line 43 of file neg_rect_lower_tri_multiply.hpp.