torch.cuda.set_stream
-
torch.cuda.set_stream(stream)[source] -
- Sets the current stream.This is a wrapper API to set the stream.
-
Usage of this function is discouraged in favor of the
streamcontext manager.
- Parameters:
-
stream (Stream) – selected stream. This function is a no-op if this argument is
None.
© 2024, PyTorch Contributors
PyTorch has a BSD-style license, as found in the LICENSE file.
https://pytorch.org/docs/1.13/generated/torch.cuda.set_stream.html
