StreamContext

Context manager for setting the current stream.

Abstract Signature:

StreamContext(stream: Stream)

PyTorch

API: torch.cuda.stream
Strategy: Direct Mapping
Official Docs ↗

Apple MLX

API: mlx.core.StreamContext
Strategy: Direct Mapping
Official Docs ↗