Back to Pytorch

Streams

docs/cpp/source/api/c10/streams.md

2.12.0636 B
Original Source

Streams

c10::Stream is the device-agnostic base stream class. It provides a common interface for working with streams across different backends (CUDA, XPU, etc.).

For backend-specific stream APIs, see {doc}../cuda/streams and {doc}../xpu/index.

Stream

{doxygenclass}
:members:
:undoc-members:

Example:

cpp
#include <c10/core/Stream.h>

// Streams are typically obtained from backend-specific APIs
auto cuda_stream = c10::cuda::getCurrentCUDAStream();

// c10::Stream provides the common interface
c10::Device device = cuda_stream.device();
c10::DeviceType type = cuda_stream.device_type();