cudaScopedDevice class
class to create an RAII-styled 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 |