TorchLean API

Docs HomeInstallationGuideExamplesGraphsTools

NN.Backend.Planner

Backend Planner #

The planner is the bridge between a semantic graph and backend capsules.

Given a kernel policy, an operation tag, and a capsule registry, it chooses an admissible capsule or explains why none is available. Graph-aware layers recover operation tags from NN.IR.OpKind and call the same selection per node.

source
structure NN.Backend.PlannedKernel :
Type

A selected kernel capsule for one graph operation.

Instances For
    source
    @[instance_reducible]
    instance NN.Backend.instReprPlannedKernel :
    Repr PlannedKernel
    source
    def NN.Backend.instReprPlannedKernel.repr :
    PlannedKernel → Nat → Std.Format
    Instances For
      source
      structure NN.Backend.KernelPlan :
      Type

      One selected kernel capsule per requested operation. This record does not execute them.

      Instances For
        source
        def NN.Backend.instReprKernelPlan.repr :
        KernelPlan → Nat → Std.Format
        Instances For
          source
          @[instance_reducible]
          instance NN.Backend.instReprKernelPlan :
          Repr KernelPlan
          source
          def NN.Backend.KernelPlan.capsuleNames (p : KernelPlan) :
          Array String

          Names of the selected backend capsules, useful for audits and logs.

          Instances For
            source
            def NN.Backend.planOp (policy : KernelPolicy) (registry : Array KernelCapsule) (op : BackendOp) :
            Except String PlannedKernel

            Choose a backend capsule for one operation.

            Instances For
              source
              def NN.Backend.planOps (policy : KernelPolicy) (registry : Array KernelCapsule) (ops : Array BackendOp) :
              Except String KernelPlan

              Choose backend capsules for a sequence of operations.

              Instances For
                source
                def NN.Backend.planOpsAvailable (policy : KernelPolicy) (availability : Availability) (registry : Array KernelCapsule) (ops : Array BackendOp) :
                Except String KernelPlan

                Choose backend capsules after filtering the registry by build availability.

                Instances For