Go to the source code of this file.
|
__kernel void | stan::math::opencl_kernels::matrix_multiply (const __global double *A, const __global double *B, __global double *C, const int M, const int N, const int K, unsigned int lower_upper_A, unsigned int lower_upper_B) |
| Matrix multiplication on the OpenCL device. More...
|
|
__kernel void | stan::math::opencl_kernels::matrix_vector_multiply (const __global double *A, const __global double *B, __global double *R, const int M, const int N, unsigned int lower_upper_A, unsigned int lower_upper_B) |
| Matrix-vector multiplication R=A*B on the OpenCL device. More...
|
|
__kernel void | stan::math::opencl_kernels::row_vector_matrix_multiply (const __global double *A, const __global double *B, __global double *R, const int N, const int K, unsigned int lower_upper_A, unsigned int lower_upper_B) |
| Row vector-matrix multiplication R=A*B on the OpenCL device. More...
|
|
|
const kernel_cl< in_buffer, in_buffer, out_buffer, int, int, int, TriangularViewCL, TriangularViewCL > | stan::math::opencl_kernels::matrix_multiply ("matrix_multiply", {thread_block_helpers, matrix_multiply_kernel_code}, {{"THREAD_BLOCK_SIZE", 32}, {"WORK_PER_THREAD", 8}}) |
| See the docs for matrix_multiply() . More...
|
|
const kernel_cl< in_buffer, in_buffer, out_buffer, int, int, TriangularViewCL, TriangularViewCL > | stan::math::opencl_kernels::matrix_vector_multiply ("matrix_vector_multiply", matrix_vector_multiply_kernel_code) |
| See the docs for matrix_vector_multiply() . More...
|
|
const kernel_cl< in_buffer, in_buffer, out_buffer, int, int, TriangularViewCL, TriangularViewCL > | stan::math::opencl_kernels::row_vector_matrix_multiply ("row_vector_matrix_multiply", row_vector_matrix_multiply_kernel_code, {{"LOCAL_SIZE_", 64}, {"REDUCTION_STEP_SIZE", 4}}) |
| See the docs for row_vector_matrix_multiply() . More...
|
|