tf::cudaScopedDevice Class Reference

class to create an RAII-styled context switch More...

#include <cuda_device.hpp>

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

Detailed Description

class to create an RAII-styled context switch

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.

Constructor & Destructor Documentation

cudaScopedDevice()

tf::cudaScopedDevice::cudaScopedDevice ( int  device )

inlineexplicit

constructs a RAII-styled device switcher

Parameters

device

device context to scope in the guard

---

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

• cuda_device.hpp