Backend Plan Audits

Inspection data for contract-carrying backend plans.

The planner chooses capsules. The audit layer records what that choice means for trust boundaries: which provider was selected, which device it targets, which spec it claims to refine, and whether the plan crosses a trusted-external boundary.

def NN.Backend.KernelCapsule.isTrustedExternal (c : KernelCapsule) : Bool

Whether this capsule crosses a trusted external boundary.

structure NN.Backend.KernelAudit : Type

Audit row for one selected backend kernel.

• op : BackendOp
• capsuleName : String
• provider : Provider
• device : Device
• specName : String
• trustLevel : TrustLevel
• vjpMode : VJPMode
• runtimeSupport : RuntimeSupport
• shapeContract : ContractDescriptor
• layoutContract : ContractDescriptor
• valueContract : ContractDescriptor
• vjpContract : ContractDescriptor

@[implicit_reducible]

instance NN.Backend.instReprKernelAudit : Repr KernelAudit

def NN.Backend.instReprKernelAudit.repr : KernelAudit → Nat → Std.Format

def NN.Backend.KernelAudit.ofPlannedKernel (k : PlannedKernel) : KernelAudit

Build an audit row from a selected planner kernel.

def NN.Backend.KernelAudit.isTrustedExternal (a : KernelAudit) : Bool

Whether this selected kernel crosses a trusted external boundary.

structure NN.Backend.ExecutionAudit : Type

Audit view of an execution plan.

• kernels : List KernelAudit

@[implicit_reducible]

instance NN.Backend.instReprExecutionAudit : Repr ExecutionAudit

def NN.Backend.instReprExecutionAudit.repr : ExecutionAudit → Nat → Std.Format

def NN.Backend.ExecutionAudit.trustLevels (a : ExecutionAudit) : List TrustLevel

Trust levels selected by the plan, in plan order.

def NN.Backend.ExecutionAudit.capsuleNames (a : ExecutionAudit) : List String

Capsule names selected by the plan, in plan order.

def NN.Backend.ExecutionAudit.trustedExternalOps (a : ExecutionAudit) : List String

Operation names whose selected capsule is trusted external.

def NN.Backend.ExecutionAudit.hasTrustedExternal (a : ExecutionAudit) : Bool

Whether the plan crosses any trusted external boundary.

def NN.Backend.ExecutionPlan.audit (p : ExecutionPlan) : ExecutionAudit

Audit a selected execution plan.

def NN.Backend.ExecutionPlan.hasTrustedExternal (p : ExecutionPlan) : Bool

Whether a selected execution plan crosses any trusted external boundary.

def NN.Backend.ExecutionPlan.trustedExternalOps (p : ExecutionPlan) : List String

Operation names whose selected capsules are trusted external.