Stan Math Library
5.0.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.