is_current_stream_capturing

paddle.device. is_current_stream_capturing ( ) [源代码] 

检查当前设备流是否处于图形捕获状态。

返回

bool: 如果当前流正在捕获则返回 True，否则返回 False。

代码示例

          >>> import paddle
>>> if paddle.device.is_available():
...     graph = paddle.device.cuda.graphs.CUDAGraph()
...     graph.capture_begin()
...     print(paddle.device.is_current_stream_capturing())  # True
...     graph.capture_end()

         