accelforge.model package#

Submodules#

accelforge.model.main module#

accelforge.model.main.evaluate_mapping(spec, flattened_arches=None, evaluated_specs=None)[source]#

Evaluate a mapping.

Parameters:

  • spec (Spec) – The specification of architecture, workload, and mapping.

  • flattened_arches (dict[(str, str), list[Leaf]] | None) – A dictionary of (EinsumName, Compute Name) to lists of architecture nodes. These contain the evaluated and flattened architecture node for that particular Einsum and compute combination. If provided, then these will be used instead of re-parsing the architecture.

  • evaluated_specs (dict[str, Spec] | None) – A dictionary of Einsum names to evaluated specifications. These contain the evaluated specification for that particular Einsum. If provided, then these will be used instead of re-parsing the specification.

Module contents#

accelforge.model.evaluate_mapping(spec, flattened_arches=None, evaluated_specs=None)[source]#

Evaluate a mapping.

Parameters:

  • spec (Spec) – The specification of architecture, workload, and mapping.

  • flattened_arches (dict[(str, str), list[Leaf]] | None) – A dictionary of (EinsumName, Compute Name) to lists of architecture nodes. These contain the evaluated and flattened architecture node for that particular Einsum and compute combination. If provided, then these will be used instead of re-parsing the architecture.

  • evaluated_specs (dict[str, Spec] | None) – A dictionary of Einsum names to evaluated specifications. These contain the evaluated specification for that particular Einsum. If provided, then these will be used instead of re-parsing the specification.