Automatic Differentiation
 
Loading...
Searching...
No Matches
multiply_transpose.hpp File Reference
#include <stan/math/opencl/kernel_cl.hpp>
#include <stan/math/opencl/buffer_types.hpp>
#include <string>

Go to the source code of this file.

Namespaces

namespace  stan
 The lgamma implementation in stan-math is based on either the reentrant safe lgamma_r implementation from C or the boost::math::lgamma implementation.
 
namespace  stan::math
 Matrices and templated mathematical functions.
 
namespace  stan::math::opencl_kernels
 

Functions

__kernel void stan::math::opencl_kernels::multiply_transpose (const __global double *A, __global double *B, const int M, const int N)
 Matrix multiplication of the form A*A^T on the OpenCL device.
 

Variables

const kernel_cl< in_buffer, out_buffer, int, int > stan::math::opencl_kernels::multiply_transpose ("multiply_transpose", {thread_block_helpers, multiply_transpose_kernel_code}, {{"THREAD_BLOCK_SIZE", 32}, {"WORK_PER_THREAD", 4}})
 See the docs for add() .