cusbf::detail::ChunkStreamPair Class Reference

Two non-blocking CUDA streams for overlapping chunk H2D and kernel work. More...

Public Member Functions
	ChunkStreamPair ()
	ChunkStreamPair (const ChunkStreamPair &)=delete
ChunkStreamPair &	operator= (const ChunkStreamPair &)=delete
	ChunkStreamPair (ChunkStreamPair &&other) noexcept
ChunkStreamPair &	operator= (ChunkStreamPair &&other) noexcept
	~ChunkStreamPair ()
cuda::stream_ref	operator[] (size_t index) const noexcept
	Non-blocking stream at index (0 or 1).
Result< void >	sync_all () const
	Blocks until both streams complete.

Detailed Description

Two non-blocking CUDA streams for overlapping chunk H2D and kernel work.

Definition at line 15 of file chunk_stream_pair.cuh.

Constructor & Destructor Documentation

ChunkStreamPair() [1/3]

cusbf::detail::ChunkStreamPair::ChunkStreamPair ( )

inline

Definition at line 17 of file chunk_stream_pair.cuh.

                      {
        for (cudaStream_t& stream : streams_) {
            CUSBF_CUDA_ABORT(cudaStreamCreateWithFlags(&stream, cudaStreamNonBlocking));
        }
    }

Here is the call graph for this function:

ChunkStreamPair() [2/3]

cusbf::detail::ChunkStreamPair::ChunkStreamPair ( const ChunkStreamPair &  )

delete

ChunkStreamPair() [3/3]

cusbf::detail::ChunkStreamPair::ChunkStreamPair ( ChunkStreamPair &&  other )

inlinenoexcept

Definition at line 26 of file chunk_stream_pair.cuh.

                                                      : streams_{other.streams_} {
        for (cudaStream_t& stream : other.streams_) {
            stream = nullptr;
        }
    }

Here is the call graph for this function:

~ChunkStreamPair()

cusbf::detail::ChunkStreamPair::~ChunkStreamPair ( )

inline

Definition at line 43 of file chunk_stream_pair.cuh.

                       {
        destroy();
    }

Member Function Documentation

operator=() [1/2]

ChunkStreamPair & cusbf::detail::ChunkStreamPair::operator= ( ChunkStreamPair &&  other )

inlinenoexcept

Definition at line 32 of file chunk_stream_pair.cuh.

                                                                 {
        if (this != &other) {
            destroy();
            streams_ = other.streams_;
            for (cudaStream_t& stream : other.streams_) {
                stream = nullptr;
            }
        }
        return *this;
    }

Here is the call graph for this function:

operator=() [2/2]

ChunkStreamPair & cusbf::detail::ChunkStreamPair::operator= ( const ChunkStreamPair &  )

delete

operator[]()

cuda::stream_ref cusbf::detail::ChunkStreamPair::operator[] ( size_t  index ) const

inlinenoexcept

Non-blocking stream at index (0 or 1).

Definition at line 48 of file chunk_stream_pair.cuh.

                                                                         {
        return {streams_.at(index)};
    }

sync_all()

Result< void > cusbf::detail::ChunkStreamPair::sync_all ( ) const

inline

Blocks until both streams complete.

Definition at line 53 of file chunk_stream_pair.cuh.

                                                {
        for (cudaStream_t stream : streams_) {
            if (stream != nullptr) {
                CUSBF_CUDA_TRY(cudaStreamSynchronize(stream));
            }
        }
        return {};
    }

Here is the call graph for this function:

---

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

• chunk_stream_pair.cuh