tf::cudaScopedDevice class
#include <src/taskflow/cuda/cuda_device.hpp>

RAII-styled device context switch.

Contents

Sample usage:

{
  tf::cudaScopedDevice device(1);  // switch to the device context 1

  // create a stream under device context 1
  cudaStream_t stream;
  cudaStreamCreate(&stream);

}  // leaving the scope and goes back to the previous device context

cudaScopedDevice is neither movable nor copyable.

Constructors, destructors, conversion operators

cudaScopedDevice(int device) explicit
constructs a RAII-styled device switcher
~cudaScopedDevice()
destructs the guard and switches back to the previous device context

Function documentation

tf::cudaScopedDevice::cudaScopedDevice(int device) explicit

constructs a RAII-styled device switcher

Parameters
device device context to scope in the guard