Skip to main content

Module execution_plan

Module execution_plan 

Source
Expand description

Pre-compiled execution plan for kernel execution.

ExecutionPlan separates one-time preparation (kernel compilation, buffer allocation) from fast repeated execution.

┌─────────────────────────────────────────────────────────┐
│              PREPARATION (one-time)                      │
│  Schedule → instantiate → compile_kernels → build()     │
│                       ↓                                  │
│                ExecutionPlan                             │
└─────────────────────────────────────────────────────────┘
                        ↓
┌─────────────────────────────────────────────────────────┐
│              EXECUTION (fast path)                       │
│  dependency-ordered PreparedOp execution                 │
└─────────────────────────────────────────────────────────┘

§Example

let plan = tensor.prepare()?;
plan.execute()?;
let output = plan.output_buffer();

Structs§

ExecutionPlan
Pre-compiled execution plan for a computation graph.
ExecutionPlanBuilder
Builder for creating ExecutionPlan from schedule data.
PreparedBufferView
Prepared zero-copy buffer view operation.
PreparedCopy
Prepared buffer-to-buffer copy operation.
PreparedCustomFunction
Prepared custom runtime function operation.
PreparedKernel
A pre-compiled kernel ready for execution.
RuntimeVar
Bound description for one DefineVar consumed by a kernel.

Enums§

PreparedOp
Prepared execution item.

Functions§

collect_runtime_vars
Walk root and collect bounds for every reachable DefineVar.