warp.capture_end# warp.capture_end(device=None, stream=None)[source]# End the capture of a graph. Parameters: device (Device | str | None) – The device where capture began stream (Stream | None) – The CUDA stream where capture began (CUDA only) Returns: A Graph object that can be launched with capture_launch() Return type: Graph
