MUDA
Loading...
Searching...
No Matches
Classes | Public Types | Public Member Functions | Static Public Member Functions | List of all members
muda::Stream Class Reference

RAII wrapper for cudaStream. More...

#include <stream.h>

Classes

class  FireAndForget
 
class  GraphFireAndForget
 
class  GraphTailLaunch
 
class  TailLaunch
 

Public Types

enum class  Flag : unsigned int { eDefault = cudaStreamDefault , eNonBlocking = cudaStreamNonBlocking }
 

Public Member Functions

MUDA_NODISCARD Stream (Flag f=Flag::eDefault)
 
 operator cudaStream_t () const
 
cudaStream_t view () const
 
 Stream (const Stream &)=delete
 
Stream & operator= (const Stream &)=delete
 
 Stream (Stream &&o) MUDA_NOEXCEPT
 
Stream & operator= (Stream &&o) MUDA_NOEXCEPT
 
void wait () const
 
void begin_capture (cudaStreamCaptureMode mode=cudaStreamCaptureModeThreadLocal) const
 
void end_capture (cudaGraph_t *graph) const
 
std::byte * workspace (size_t byte_size)
 

Static Public Member Functions

static Stream & Default ()
 

Detailed Description

RAII wrapper for cudaStream.

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