Struct StreamGraphCapture 

pub struct StreamGraphCapture<'s> { /* private fields */ }

Driver-backed CUDA stream capture.

Where StreamCapture is a CPU-side recorder (it only logs operation specifications), StreamGraphCapture drives the real cuStreamBeginCapture_v2 / cuStreamEndCapture API. The caller begins capture on a live Stream, submits ordinary GPU work to that stream (kernel launches, async memset/memcpy) which the driver records instead of executing, and end finalises the captured CUgraph into a launchable GraphExec. Launching that exec replays the captured work.

Requires a driver with stream-capture support (CUDA 10.0+); otherwise begin returns CudaError::NotSupported.

Implementations

impl<'s> StreamGraphCapture<'s>

pub fn begin(stream: &'s Stream, mode: CUstreamCaptureMode) -> CudaResult<Self>

Begins driver-backed capture on stream with the given capture mode (e.g. CU_STREAM_CAPTURE_MODE_GLOBAL).

Errors

• CudaError::NotInitialized when no driver is loaded.
• CudaError::NotSupported when the driver predates stream capture.
• Any CudaError mapped from cuStreamBeginCapture_v2.

pub fn is_active(&self) -> bool

Returns whether the capture is still active (not yet ended).

pub fn capture_status(&self) -> CudaResult<CUstreamCaptureStatus>

Reports the driver’s capture status for the stream (cuStreamIsCapturing).

Errors

Propagates driver-load failures and any error from cuStreamIsCapturing.

pub fn end(self) -> CudaResult<GraphExec>

Ends capture and instantiates the captured CUgraph into a launchable GraphExec.

The returned exec’s node_count reflects the number of nodes the driver actually captured (queried via cuGraphGetNodes), and launch replays the captured work via cuGraphLaunch.

Errors

Propagates driver-load failures and any error from cuStreamEndCapture or graph instantiation. On instantiation failure the captured graph is destroyed before returning so the driver object does not leak.

Trait Implementations

impl Drop for StreamGraphCapture<'_>

fn drop(&mut self)

Executes the destructor for this type.

fn pin_drop(self: Pin<&mut Self>)

🔬This is a nightly-only experimental API. (pin_ergonomics)

Execute the destructor for this type, but different to Drop::drop, it requires self to be pinned.