torch.mps.stream

torch.mps.stream(stream)

Context manager that selects a given stream.

MPS operators called within this context will be enqueued on the selected stream.

Parameters:

stream (Stream) – selected stream. This manager is a no-op if it’s None.