planner

AotMemoryPlanner API

This class is the base class for all memory planners in the HeliaAOT framework.

Classes:

• AotMemoryPlanner –

  Base class for memory planners

Functions:

Copyright 2025 Ambiq. All Rights Reserved.

Classes

AirMemoryPlanner

AirMemoryPlanner(target: SocPlatform)

Functions

plan

plan(model: AirModel, memory_constraints: list[MemoryConstraint] | None = None, tensor_constraints: list[AttributeRuleset] | None = None) -> MemoryPlan

Plan tensor memory allocation for the model.

Parameters:

• model

  (AirModel) –

  The target model.

• memory_constraints

  (list[MemoryConstraint] | None, default: None ) –

  Memory constraints

• tensor_constraints

  (list[AttributeRuleset] | None, default: None ) –

  Tensor constraints

Returns:

• MemoryPlan ( MemoryPlan ) –

  The memory plan containing offsets and total size.

Functions

register_memory_planner

register_memory_planner(cls: type[AirMemoryPlanner]) -> type[AirMemoryPlanner]

Class decorator: register each memory planner by its NAME.

get_memory_planner

get_memory_planner(name: str) -> type[AirMemoryPlanner]

Get a memory planner by its name.