set_stream¶

paddle.device. set_stream ( stream ) [source]

Set the current stream. :param stream: The selected stream. :type stream: Stream

Returns: The previous stream.
Return type: Stream

Examples

System Message: ERROR/3 (/usr/local/lib/python3.8/site-packages/paddle/device/__init__.py:docstring of paddle.device.set_stream, line 10)

Error in “code-block” directive: maximum 1 argument(s) allowed, 6 supplied.

            .. code-block:: python
    # required: custom_device
    import paddle

    paddle.set_device('custom_cpu')
    s = paddle.device.Stream()
    paddle.device.set_stream(s)

           