![]() |
Stan Math Library
5.1.0
Automatic Differentiation
|
| __kernel void stan::math::opencl_kernels::tridiagonalization_v_step_1 | ( | const __global double * | P, |
| const __global double * | V, | ||
| __global double * | Uu, | ||
| __global double * | Vu, | ||
| const int | P_rows, | ||
| const int | V_rows, | ||
| const int | k | ||
| ) |
Calculates first part of constructing the vector v: Uu = Pb * u and Vu = Vl * u.
Pb is a block of packed matrix, Vl is left part of matrix V and u is householder vector. Must be run with number of work groups equal to size of resulting vectors and 64 threads per work group.
| P | Packed matrix being constructed. | |
| V | Matrix V. | |
| [out] | Uu | First resulting vector. |
| [out] | Vu | Second resulting vector. |
| P_rows | Number of rows of the packed matrix | |
| V_rows | Number of rows of the matrix V | |
| k | Index of the householder vector in the block we use as input |
Definition at line 119 of file tridiagonalization.hpp.