Backend Planning for IR Graphs

Adapter from TorchLean's op-tagged IR to backend capsules.

This is intentionally a planning layer, not an evaluator. The graph remains the semantic object; the planner selects which backend contract should implement each node or node family.

def NN.Backend.IR.op? : IR.OpKind → Option BackendOp

Backend operation tag for an IR op.

These tags are the same vocabulary used by backend capsules and runtime reports. If an op maps to a tag that has no capsule for the selected profile, planning fails at the backend boundary instead of silently using a broader bucket.

def NN.Backend.IR.nodeOp? (n : IR.Node) : Option BackendOp

Backend operation requested by a graph node, if the node needs runtime work.

structure NN.Backend.IR.PlannedNodeKernel : Type

Backend choice for one concrete IR node.

• nodeId : ℕ
• kind : IR.OpKind
• op : BackendOp
• capsule : KernelCapsule

def NN.Backend.IR.instReprPlannedNodeKernel.repr : PlannedNodeKernel → ℕ → Std.Format

@[implicit_reducible]

instance NN.Backend.IR.instReprPlannedNodeKernel : Repr PlannedNodeKernel

structure NN.Backend.IR.GraphExecutionPlan : Type

Graph-aware execution plan that preserves the IR node identity for every backend choice.

• kernels : List PlannedNodeKernel

def NN.Backend.IR.instReprGraphExecutionPlan.repr : GraphExecutionPlan → ℕ → Std.Format

@[implicit_reducible]

instance NN.Backend.IR.instReprGraphExecutionPlan : Repr GraphExecutionPlan

def NN.Backend.IR.GraphExecutionPlan.nodeIds (p : GraphExecutionPlan) : List ℕ

Node ids covered by backend kernels, in graph order.

def NN.Backend.IR.GraphExecutionPlan.capsuleNames (p : GraphExecutionPlan) : List String

Selected capsule names, in graph order.

def NN.Backend.IR.planNode? (cfg : ExecutionConfig) (availability : Availability) (registry : List KernelCapsule) (n : IR.Node) : Except String (Option PlannedNodeKernel)

Plan a single IR node when it corresponds to runtime work.

def NN.Backend.IR.planGraphNodesWithRegistry (cfg : ExecutionConfig) (availability : Availability) (registry : List KernelCapsule) (g : IR.Graph) : Except String GraphExecutionPlan

Plan every runtime-relevant node in graph order.

def NN.Backend.IR.checkedPlanGraphNodesWithRegistry (cfg : ExecutionConfig) (availability : Availability) (registry : List KernelCapsule) (g : IR.Graph) : Except String GraphExecutionPlan

Check graph well-formedness, then plan every runtime-relevant node.