Stan Math Library
5.0.0
Automatic Differentiation
|
Creates functor for kernels.
Args | Parameter pack of all kernel argument types. |
Definition at line 174 of file kernel_cl.hpp.
#include <kernel_cl.hpp>
Public Member Functions | |
kernel_cl (const char *name, std::vector< std::string > sources, std::unordered_map< std::string, int > options={}) | |
Creates functor for kernels that only need access to defining the global work size. | |
template<typename... CallArgs> | |
auto | operator() (cl::NDRange global_thread_size, CallArgs &&... args) const |
Executes a kernel. | |
template<typename... CallArgs> | |
auto | operator() (cl::NDRange global_thread_size, cl::NDRange thread_block_size, CallArgs &&... args) const |
Executes a kernel. | |
int | get_option (const std::string option_name) const |
Retrieves an option used for compiling the kernel. | |
Private Attributes | |
const char * | name_ |
std::vector< std::string > | sources_ |
std::unordered_map< std::string, int > | opts_ |
cl::Kernel | kernel_ |