torch.accelerator.current_stream¶
- torch.accelerator.current_stream(device=None, /)[source][source]¶
Return the currently selected stream for a given device.
- Parameters
device (
torch.device
, str, int, optional) – a given device that must match the current accelerator device type. If not given, usetorch.accelerator.current_device_idx()
by default.- Returns
the currently selected stream for a given device.
- Return type