# -*- coding: utf-8 -*-
# (C) Copyright 2020, 2021, 2022, 2023, 2024 IBM. All Rights Reserved.
#
# Licensed under the MIT license. See LICENSE file in the project root for details.
# pylint: disable=too-many-instance-attributes
# pylint: disable=too-many-lines
"""Runtime parameters for settings at simulation time."""
from dataclasses import dataclass
from .helpers import _PrintableMixin
from .enums import RPUDataType
[docs]
@dataclass
class RuntimeParameter(_PrintableMixin):
"""Parameter that define the analog-matvec (forward / backward) and
peripheral digital input-output behavior.
Here one can enable analog-digital conversion, dynamic input
scaling, and define the properties of the analog-matvec
computations, such as noise and non-idealities (e.g. IR-drop).
"""
data_type: RPUDataType = RPUDataType.FLOAT
"""Data type to use for the C++ bindings and simulations.
"""
on_the_fly_bindings: bool = False
"""Enable on the fly generation of some parameter bindings.
For some post update steps (e.g. for inference tile), parameter
bindings are generated on the fly. This is only useful if these
parameters are vary during training and is turned off by default
for performance reason (which typically is not supported as the
`RPUConfig` is considered fixed once the analog tile is
instantiated).
If indeed parameters (e.g. ``WeightClipParameter``,
``WeightModifierParameter``, and ``WeightRemapParameter``) vary
during training, this options as to be turned off to re-generate
the bindings on the fly (that is each mini-batch).
"""
offload_input: bool = False
"""Whether to offload the stored input for the update pass on
CPU to save GPU memory.
"""
offload_gradient: bool = False
"""Whether to offload the stored gradient for the update pass on
CPU to save GPU memory.
Note:
Only for in case tiles are simulated with RPUCuda library.
"""