Enum NodeKind 

pub enum NodeKind {
    Assign {
        name: String,
        dtype: DType,
        dims: Vec<String>,
    },
    Op {
        op: OpKind,
        attrs: OpAttrs,
        inputs: Vec<String>,
        output: String,
    },
    Branch {
        cond: Option<String>,
        then_block: String,
        else_block: Option<String>,
    },
    Barrier,
    Dep {
        after: String,
        before: String,
    },
    CacheRead {
        src: CacheAccess,
        dst: String,
    },
    CacheWrite {
        src: String,
        dst: CacheAccess,
    },
    CacheIncrement {
        target: String,
        amount: i64,
    },
    CacheDecrement {
        target: String,
        amount: i64,
    },
    CacheReset {
        target: CacheAccess,
    },
    Transfer {
        src: String,
        dst: String,
    },
    Yield {
        vars: Vec<String>,
    },
    Await {
        vars: Vec<String>,
    },
    Loop {
        name: String,
        index: String,
        start: String,
        end: String,
        body: Vec<Node>,
    },
    Return,
}

Node variants that make up a graph.

Variants

Assign

Fields

name: String

dtype: DType

dims: Vec<String>

Op

Fields

op: OpKind

attrs: OpAttrs

inputs: Vec<String>

output: String

Branch

Fields

cond: Option<String>

then_block: String

else_block: Option<String>

Barrier

Dep

Fields

after: String

before: String

CacheRead

Fields

src: CacheAccess

dst: String

CacheWrite

Fields

src: String

dst: CacheAccess

CacheIncrement

Fields

target: String

amount: i64

CacheDecrement

Fields

target: String

amount: i64

CacheReset

Fields

target: CacheAccess

Transfer

Fields

src: String

dst: String

Yield

Fields

vars: Vec<String>

Await

Fields

vars: Vec<String>

Loop

Fields

name: String

index: String

start: String

end: String

body: Vec<Node>

Return

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<'de> Deserialize<'de> for NodeKind

fn deserialize<__D>(__deserializer: __D) -> Result<Self, __D::Error>

where __D: Deserializer<'de>,

Deserialize this value from the given Serde deserializer.

impl Serialize for NodeKind

fn serialize<__S>(&self, __serializer: __S) -> Result<__S::Ok, __S::Error>

where __S: Serializer,

Serialize this value into the given Serde serializer.