warp.capture_end

warp.capture_end(device=None, stream=None)

End the capture of a CUDA graph.

Parameters:

• device (Device | str | None) – The CUDA device where capture began

• stream (Stream | None) – The CUDA stream where capture began

Returns:

A Graph object that can be launched with capture_launch()

Return type:

Graph