aihwkit.cloud.converter.v1.rpu_config_info module

Creates InferenceRPUConfig to add to nn model

exception aihwkit.cloud.converter.v1.rpu_config_info.NoiseModelDeviceIDException[source]

Bases: Exception

Exception raised if noise model device id is not correct

class aihwkit.cloud.converter.v1.rpu_config_info.RPUconfigInfo(nm_info, a_info, layers=None)[source]

Bases: object

Data only class for RPUConfig fields

Parameters:
create_inference_rpu_config(func_id, verbose=False)[source]

Creates a InferenceRPUConfig class using noise and analog info

Parameters:

  • func_id (str) –

  • verbose (bool) –

Return type:

InferenceRPUConfig