Stan Math Library  2.20.0
reverse mode automatic differentiation
Namespaces | Functions | Variables
matrix_multiply.hpp File Reference
#include <stan/math/opencl/kernel_cl.hpp>
#include <stan/math/opencl/buffer_types.hpp>

Go to the source code of this file.

Namespaces

 stan
 
 stan::math
 Matrices and templated mathematical functions.
 
 stan::math::opencl_kernels
 

Functions

__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...
 

Variables

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...
 

     [ Stan Home Page ] © 2011–2018, Stan Development Team.