Enum NodeKind 

pub enum NodeKind {
    KernelLaunch {
        function_name: String,
        config: KernelConfig,
        fusible: bool,
    },
    Memcpy {
        dir: MemcpyDir,
        size_bytes: usize,
    },
    Memset {
        size_bytes: usize,
        value: u8,
    },
    EventRecord,
    EventWait,
    HostCallback {
        label: String,
    },
    Barrier,
    Conditional {
        condition_buf: BufferId,
        true_branch: Vec<NodeId>,
        false_branch: Vec<NodeId>,
    },
}

The operation kind associated with a GraphNode.

Variants

KernelLaunch

Launches a PTX/CUDA kernel.

Fields

function_name: String

Kernel function name (used to look up the loaded module).

config: KernelConfig

Grid/block configuration.

fusible: bool

Whether this kernel is eligible for fusion with adjacent element-wise kernels.

Memcpy

Copies memory between host and device, or device to device.

Fields

dir: MemcpyDir

Transfer direction.

size_bytes: usize

Transfer size in bytes.

Memset

Fills a device buffer with a byte pattern.

Fields

size_bytes: usize

Number of bytes to fill.

value: u8

Byte fill value.

EventRecord

Records a CUDA event (signals completion of preceding work).

EventWait

Waits for a recorded event before proceeding.

HostCallback

A host-side callback inserted into the stream.

Fields

label: String

Human-readable label for debugging and visualisation.

Barrier

A no-op synchronisation barrier (join point for multiple dependency chains).

Conditional

A conditional subgraph fork (CUDA graph conditional node equivalent).

Fields

condition_buf: BufferId

Condition variable — a boolean device buffer.

true_branch: Vec<NodeId>

Which branch is the “true” path (node IDs within this graph).

false_branch: Vec<NodeId>

Which branch is the “false” path.

Implementations

impl NodeKind

pub fn is_compute(&self) -> bool

Returns true if this node performs device computation (not just memory movement).

pub fn is_memory_op(&self) -> bool

Returns true if this node is a memory transfer.

pub fn is_fusible(&self) -> bool

Returns true if this kernel is marked as fusible with adjacent element-wise kernels.

pub fn function_name(&self) -> Option<&str>

Returns the function name for kernel launch nodes.

pub fn tag(&self) -> &'static str

Returns a short human-readable tag for this operation kind.

Trait Implementations

impl Clone for NodeKind

fn clone(&self) -> NodeKind

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for NodeKind

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

Formats the value using the given formatter.

impl Display for NodeKind

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

Formats the value using the given formatter.

impl PartialEq for NodeKind

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

Equality operator ==.

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

Inequality operator !=.

impl StructuralPartialEq for NodeKind