MUDA
Loading...
Searching...
No Matches
Public Member Functions | Public Attributes | Protected Member Functions | List of all members
muda::CubWrapper< Derive > Class Template Reference
Inheritance diagram for muda::CubWrapper< Derive >:
muda::LaunchBase< Derive > muda::LaunchCore

Public Member Functions

 CubWrapper (Stream &stream=Stream::Default())
 
void kernel_name (std::string_view)=delete
 
- Public Member Functions inherited from muda::LaunchBase< Derive >
MUDA_GENERIC LaunchBase (::cudaStream_t stream) MUDA_NOEXCEPT
 
MUDA_GENERIC LaunchBase (cudaStream_t stream) MUDA_NOEXCEPT
 
Derive & push_range (const std::string &name)
 
Derive & pop_range ()
 
Derive & kernel_name (std::string_view name)
 
Derive & file_line (std::string_view file, int line)
 
Derive & record (cudaEvent_t e, int flag=cudaEventRecordDefault)
 
Derive & record (ComputeGraphVar< cudaEvent_t > &e, const std::vector< ComputeGraphVarBase * > &vars)
 
Derive & record (ComputeGraphVar< cudaEvent_t > &e, ComputeGraphVar< ViewT > &... vars)
 
Derive & when (cudaEvent_t e, int flag=cudaEventWaitDefault)
 
Derive & wait (cudaEvent_t e, int flag=cudaEventWaitDefault)
 
Derive & wait (const ComputeGraphVar< cudaEvent_t > &e, const std::vector< ComputeGraphVarBase * > &vars)
 
Derive & wait (const ComputeGraphVar< cudaEvent_t > &e, ComputeGraphVar< ViewT > &... vars)
 
Derive & wait ()
 
Derive & callback (const std::function< void(::cudaStream_t, ::cudaError)> &callback)
 
Next next (Next n)
 
Next next (Args &&... args)
 
- 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)
 

Public Attributes

Stream * m_muda_stream = nullptr
 

Protected Member Functions

std::byte * prepare_buffer (size_t reqSize)
 
- Protected Member Functions inherited from muda::LaunchBase< Derive >
Derive & pop_kernel_label ()
 
- Protected Member Functions inherited from muda::LaunchCore
MUDA_GENERIC::cudaStream_t stream () const
 
MUDA_HOST void pop_kernel_label ()
 

Additional Inherited Members

- Public Types inherited from muda::LaunchBase< Derive >
using derived_type = Derive
 
- Static Public Member Functions inherited from muda::LaunchCore
static void kernel_name (std::string_view name)
 
static void file_line (std::string_view file, int line)
 
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 file: