class

ML::Metal::ComputeGraph

Constructors

new

debug_encode

debug_encode=(debug_encode : Bool)

add_op(pipeline : ComputePipeline, &block : OpBuilder -> ) : Int32

Add an operation to the graph. Returns its index.

add_op_direct(op : Op) : Int32

Add a pre-built Op directly (used by GraphEncoder)

clear

Clear all operations (reuse the graph object)

compile!

Compile: analyze dependencies, group into waves

encode(cmd : CommandBuffer) : Nil

Encode the compiled graph into a command buffer

encode_sequential(cmd : CommandBuffer) : Nil

Encode sequentially (debug: barrier between every op, no wave optimization)

size

Number of operations in the graph

stats

Compilation stats (available after compile!)