Enum CaptureMode 

pub enum CaptureMode {
    Global,
    ThreadLocal,
    Relaxed,
}

Selects how CUDA graph capture serializes interactions with other threads. Mirrors cudaStreamCaptureMode.

• Global — any CUDA API call from any thread that touches the capturing stream (or any thread that is also capturing) will invalidate capture. Safest for debugging; matches PyTorch’s default.
• ThreadLocal — only calls from the capturing thread can invalidate capture. Other threads may freely use unrelated streams. This is what ferrotorch-gpu has always used.
• Relaxed — the driver does not track cross-thread interactions at all. Fastest, but the caller is fully responsible for making sure no other thread interferes.

Variants

Global

Global serialization (CU_STREAM_CAPTURE_MODE_GLOBAL).

ThreadLocal

Thread-local serialization (CU_STREAM_CAPTURE_MODE_THREAD_LOCAL). This is the default in PyTorch’s cuda.graph context.

Relaxed

Relaxed — no cross-thread serialization (CU_STREAM_CAPTURE_MODE_RELAXED).

Implementations

impl CaptureMode

pub fn to_cuda(self) -> CUstreamCaptureMode

Convert to the raw cudarc enum.

Trait Implementations

impl Clone for CaptureMode

fn clone(&self) -> CaptureMode

Returns a duplicate of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl Debug for CaptureMode

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.

impl Default for CaptureMode

fn default() -> Self

Returns the “default value” for a type.

impl Hash for CaptureMode

fn hash<__H: Hasher>(&self, state: &mut __H)

Feeds this value into the given Hasher.

fn hash_slice<H>(data: &[Self], state: &mut H)

where H: Hasher, Self: Sized,

Feeds a slice of this type into the given Hasher.

impl PartialEq for CaptureMode

fn eq(&self, other: &CaptureMode) -> bool

Tests for self and other values to be equal, and is used by ==.

fn ne(&self, other: &Rhs) -> bool

Tests for !=. The default implementation is almost always sufficient, and should not be overridden without very good reason.

impl Copy for CaptureMode

impl Eq for CaptureMode

impl StructuralPartialEq for CaptureMode