Automatic Differentiation
 
Loading...
Searching...
No Matches
stan::math::opencl_kernels::kernel_cl< Args > Struct Template Reference

Detailed Description

template<typename... Args>
struct stan::math::opencl_kernels::kernel_cl< Args >

Creates functor for kernels.

Template Parameters
ArgsParameter 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_
 

The documentation for this struct was generated from the following file: