Ginkgo: gko::cuda_stream Class Reference

| Ginkgo Generated from branch based on main. Ginkgo version 1.11.0 A numerical linear algebra library targeting many-core architectures

Loading...
Searching...
No Matches
gko::cuda_stream Class Reference

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

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

## Public Member Functions
| 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.
| ~cuda_stream ()
| Destroys the custom CUDA stream, if it isn't empty.
| cuda_stream (const cuda_stream &)=delete
| cuda_stream (cuda_stream &&)
| Move-constructs from an existing stream, which will be emptied.
cuda_stream & | operator= (const cuda_stream &)=delete
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.

## Detailed Description

An RAII wrapper for a custom CUDA stream.

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

## Constructor & Destructor Documentation

## ◆ cuda_stream()

gko::cuda_stream::cuda_stream | ( | int | device_id | ) |

Creates a new custom CUDA stream on the given device.

Parameters

  device_id | the device ID to create the stream on.

## Member Function Documentation

## ◆ get()

CUstream_st * gko::cuda_stream::get | ( |  | ) | const

Returns the native CUDA stream handle.

In an empty cuda_stream, this will return nullptr.

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

- ginkgo/core/base/stream.hpp