# StreamContext

Source: https://www.tensorplay.cn/docs/generated/tensorplay.cuda.StreamContext.html

# StreamContext

class tensorplay.cuda.StreamContext(stream: [Stream](tensorplay.cuda.streams.Stream.html#tensorplay.cuda.streams.Stream) | [None](https://docs.python.org/3/library/constants.html#None))[[source]](../_modules/tensorplay/cuda.html#StreamContext)

Context-manager that selects a given stream.

All CUDA kernels queued within its context will be enqueued on a selected
stream.

Parameters:

Stream ([Stream](tensorplay.cuda.streams.Stream.html#tensorplay.cuda.streams.Stream)) – selected stream. This manager is a no-op if it’s
None.

Note

Streams are per-device.
