Shortcuts

torch.accelerator.set_stream

torch.accelerator.set_stream(stream)[source]

Set the current stream to a given stream.

Parameters

stream (torch.Stream) – a given stream that must match the current accelerator device type.

Note

This function will set the current device index to the device index of the given stream.

Docs

Access comprehensive developer documentation for PyTorch

View Docs

Tutorials

Get in-depth tutorials for beginners and advanced developers

View Tutorials

Resources

Find development resources and get your questions answered

View Resources