cuda::experimental::stf::cuda_kernel_desc

struct cuda_kernel_desc

Description of a CUDA kernel.

This is used to describe kernels passed to the ctx.cuda_kernel and ctx.cuda_kernel_chain API calls.

Public Types

using func_variant_t = ::std::variant<CUfunction, CUkernel, const void*>

Public Functions

cuda_kernel_desc() = default

template<typename Fun, typename ...Args>
inline cuda_kernel_desc(

Fun func,

dim3 gridDim_,

dim3 blockDim_,

size_t sharedMem_,

Args... args,

)

template<typename Fun, typename ...Args>
inline void configure(

Fun func,

dim3 gridDim_,

dim3 blockDim_,

size_t sharedMem_,

Args... args,

)

template<typename Fun>
inline void configure_raw(

Fun func,

dim3 gridDim_,

dim3 blockDim_,

size_t sharedMem_,

int arg_cnt,

const void **args,

)

inline void launch(cudaStream_t stream) const

inline void launch_in_graph(

cudaGraphNode_t &node,

cudaGraph_t &graph,

) const

inline int get_num_registers() const

Public Members

func_variant_t func_variant

dim3 gridDim

dim3 blockDim

size_t sharedMem = 0

mutable ::std::vector<void*> args_ptr