5#ifndef GKO_PUBLIC_CORE_BASE_STREAM_HPP_
6#define GKO_PUBLIC_CORE_BASE_STREAM_HPP_
9#include <ginkgo/core/base/executor.hpp>
49 CUstream_st*
get()
const;
92 GKO_HIP_STREAM_STRUCT*
get()
const;
95 GKO_HIP_STREAM_STRUCT* stream_;
An RAII wrapper for a custom CUDA stream.
Definition stream.hpp:20
~cuda_stream()
Destroys the custom CUDA stream, if it isn't empty.
cuda_stream(cuda_stream &&)
Move-constructs from an existing stream, which will be emptied.
cuda_stream & operator=(cuda_stream &&)=delete
Move-assigns from an existing stream, which will be emptied.
CUstream_st * get() const
Returns the native CUDA stream handle.
cuda_stream()
Creates an empty stream wrapper, representing the default stream.
cuda_stream(int device_id)
Creates a new custom CUDA stream on the given device.
An RAII wrapper for a custom HIP stream.
Definition stream.hpp:63
hip_stream(hip_stream &&)
Move-constructs from an existing stream, which will be emptied.
hip_stream & operator=(hip_stream &&)=delete
Move-assigns from an existing stream, which will be emptied.
hip_stream(int device_id)
Creates a new custom HIP stream on the given device.
CUstream_st * get() const
Returns the native HIP stream handle.
hip_stream()
Creates an empty stream wrapper, representing the default stream.
~hip_stream()
Destroys the custom HIP stream, if it isn't empty.
The Ginkgo namespace.
Definition abstract_factory.hpp:20