muda
Public Types | Public Member Functions | Protected Member Functions | Friends | List of all members
muda::LaunchBase< T > Class Template Reference
Inheritance diagram for muda::LaunchBase< T >:
Inheritance graph
[legend]
Collaboration diagram for muda::LaunchBase< T >:
Collaboration graph
[legend]

Public Types

using derived_type = T
 

Public Member Functions

MUDA_GENERIC LaunchBase (::cudaStream_t stream) MUDA_NOEXCEPT
 
T & push_range (const std::string &name)
 
T & pop_range ()
 
T & kernel_name (std::string_view name)
 
std::string_view kernel_name () const
 
T & record (cudaEvent_t e, int flag=cudaEventRecordDefault)
 
T & record (ComputeGraphVar< cudaEvent_t > &e, const std::vector< ComputeGraphVarBase *> &vars)
 
template<typename... ViewT>
T & record (ComputeGraphVar< cudaEvent_t > &e, ComputeGraphVar< ViewT > &... vars)
 
T & when (cudaEvent_t e, int flag=cudaEventWaitDefault)
 
T & wait (cudaEvent_t e, int flag=cudaEventWaitDefault)
 
T & wait (const ComputeGraphVar< cudaEvent_t > &e, const std::vector< ComputeGraphVarBase *> &vars)
 
template<typename... ViewT>
T & wait (const ComputeGraphVar< cudaEvent_t > &e, ComputeGraphVar< ViewT > &... vars)
 
T & wait ()
 
T & callback (const std::function< void(::cudaStream_t, ::cudaError)> &callback)
 
template<typename Next >
Next next (Next n)
 
template<typename Next , typename... Args>
Next next (Args &&... args)
 
template<typename T >
MUDA_GENERIC LaunchBase (cudaStream_t stream) MUDA_NOEXCEPT
 
- Public Member Functions inherited from muda::LaunchCore
MUDA_GENERIC LaunchCore (::cudaStream_t stream) MUDA_NOEXCEPT
 
void init_stream (::cudaStream_t s)
 
void push_range (const std::string &name)
 
void pop_range ()
 
void record (cudaEvent_t e, int flag=cudaEventRecordDefault)
 
void record (ComputeGraphVar< cudaEvent_t > &e, const std::vector< ComputeGraphVarBase *> &vars)
 
template<typename... ViewT>
void record (ComputeGraphVar< cudaEvent_t > &e, ComputeGraphVar< ViewT > &... vars)
 
void when (cudaEvent_t e, int flag=cudaEventWaitDefault)
 
void wait (cudaEvent_t e, int flag=cudaEventWaitDefault)
 
void wait (const ComputeGraphVar< cudaEvent_t > &e, const std::vector< ComputeGraphVarBase *> &vars)
 
template<typename... ViewT>
void wait (const ComputeGraphVar< cudaEvent_t > &e, ComputeGraphVar< ViewT > &... vars)
 
void wait ()
 
void callback (const std::function< void(::cudaStream_t, ::cudaError)> &callback)
 
template<typename... ViewT>
MUDA_INLINE void record (ComputeGraphVar< cudaEvent_t > &e, ComputeGraphVar< ViewT > &... vars)
 
template<typename... ViewT>
MUDA_INLINE void wait (const ComputeGraphVar< cudaEvent_t > &e, ComputeGraphVar< ViewT > &... vars)
 

Protected Member Functions

T & pop_kernel_name ()
 
- Protected Member Functions inherited from muda::LaunchCore
MUDA_GENERIC ::cudaStream_t stream () const
 
MUDA_HOST void pop_kernel_name ()
 

Friends

template<typename Others >
class LaunchBase
 

Additional Inherited Members

- Static Public Member Functions inherited from muda::LaunchCore
static void kernel_name (std::string_view name)
 
static std::string_view kernel_name ()
 
static void wait_event (cudaEvent_t event)
 
static void wait_stream (::cudaStream_t stream)
 
static void wait_device ()
 
- Protected Types inherited from muda::LaunchCore
template<typename T >
using S = std::shared_ptr< T >
 
- Protected Attributes inherited from muda::LaunchCore
::cudaStream_t m_stream
 

The documentation for this class was generated from the following files: