Ginkgo  Generated from pipelines/1554403166 branch based on develop. Ginkgo version 1.9.0
A numerical linear algebra library targeting many-core architectures
Public Member Functions | List of all members
gko::hip_stream Class Reference

An RAII wrapper for a custom HIP stream. More...

#include <ginkgo/core/base/stream.hpp>

Public Member Functions

 hip_stream ()
 Creates an empty stream wrapper, representing the default stream.
 
 hip_stream (int device_id)
 Creates a new custom HIP stream on the given device. More...
 
 ~hip_stream ()
 Destroys the custom HIP stream, if it isn't empty.
 
 hip_stream (const hip_stream &)=delete
 
 hip_stream (hip_stream &&)
 Move-constructs from an existing stream, which will be emptied.
 
hip_streamoperator= (const hip_stream &)=delete
 
hip_streamoperator= (hip_stream &&)=delete
 Move-assigns from an existing stream, which will be emptied.
 
CUstream_st * get () const
 Returns the native HIP stream handle. More...
 

Detailed Description

An RAII wrapper for a custom HIP stream.

The stream will be created on construction and destroyed when the lifetime of the wrapper ends.

Constructor & Destructor Documentation

◆ hip_stream()

gko::hip_stream::hip_stream ( int  device_id)

Creates a new custom HIP stream on the given device.

Parameters
device_idthe device ID to create the stream on.

Member Function Documentation

◆ get()

CUstream_st* gko::hip_stream::get ( ) const

Returns the native HIP stream handle.

In an empty hip_stream, this will return nullptr.


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