Devices and Streams#
A device to run operations on. |
|
A stream for running operations on a given device. |
|
Get the default device. |
|
|
Set the default device. |
|
Get the device's default stream. |
|
Make a new stream on the given device. |
|
Set the default stream. |
|
Create a context manager to set the default device and stream. |
|
Synchronize with the given stream. |