torch.mps.set_stream

torch.mps.set_stream(stream)

Set the current stream. Usage of this function is discouraged in favor of the stream() context manager.

Parameters:

stream (Stream) – selected stream. This function is a no-op if this argument is None.