aihwkit_lightning.simulator.parameters.mapping module

Mapping parameters for resistive processing units.

class aihwkit_lightning.simulator.parameters.mapping.MappableRPU(mapping=<factory>)[source]

Bases: _PrintableMixin

Defines the mapping parameters and utility factories

mapping: MappingParameter: Parameter related to mapping weights to tiles for supporting modules.

class aihwkit_lightning.simulator.parameters.mapping.MappingParameter(max_input_size=-1)[source]

Bases: _PrintableMixin

Parameter related to hardware design and the mapping of logical weight matrices to physical tiles.

Caution

Some of these parameters have only an effect for modules that support tile mappings.

max_input_size: int = -1

Maximal input size of the weight matrix that is handled on a single analog tile.

If the logical weight matrix size exceeds this size it will be split and mapped onto multiple analog tiles. -1 means no maximum input size.