Support for graded spikes in pre-synaptic trace updates

* use of graded spike payload to update pre-synaptic trace (bit-acc) * progress in unit testing graded spike in learning feature * progress in unit testing graded spike in learning feature * minor changes * updated documentation * more tests * finished testing * adding missing license headers * fixing linting issues * addressing change requests * improved tests * fixing lint * improved tests * updating license headers * Test ulimit change * test ulimit change * Update ci.yml * Update ci.yml * Update ci.yml * Update ci.yml * Update ci.yml * Update ci.yml * Update test_models.py * Update test_models.py * Update ci.yml * Update ci.yml * Update ci.yml * Update ci.yml * Update ci.yml * Update ci.yml * Update ci.yml * Update ci.yml * Update ci.yml * Update ci.yml * Update ci.yml * Update ci.yml * improving code quality --------- Co-authored-by: gkarray <[email protected]> Co-authored-by: PhilippPlank <[email protected]> Co-authored-by: Mathis Richter <[email protected]>
lava-nc · Feb 21, 2023 · f209c56 · f209c56
1 parent 359533b
commit f209c56
Show file tree

Hide file tree

Showing 6 changed files with 1,520 additions and 182 deletions.
diff --git a/.github/workflows/ci.yml b/.github/workflows/ci.yml
@@ -67,4 +67,5 @@ jobs:
       run: |
         poetry run git lfs fetch
         poetry run git lfs pull
-        poetry run pytest
+        # poetry run pytest
+        poetry run python -m unittest discover -s tests/ -t . -vv
diff --git a/src/lava/magma/core/learning/constants.py b/src/lava/magma/core/learning/constants.py
@@ -1,7 +1,8 @@
-# Copyright (C) 2021-22 Intel Corporation
+# Copyright (C) 2021-23 Intel Corporation
 # SPDX-License-Identifier: BSD-3-Clause
 # See: https://spdx.org/licenses/
 
+from enum import IntEnum, auto
 import lava.magma.core.learning.string_symbols as str_symbols
 
 # ---------------------------------------------------------------------------
@@ -85,3 +86,10 @@
     str_symbols.Y2: 1,
     str_symbols.Y3: 2
 }
+
+
+class GradedSpikeCfg(IntEnum):
+    USE_REGULAR_IMPULSE = auto()
+    OVERWRITE = auto()
+    ADD_WITH_SATURATION = auto()
+    ADD_WITHOUT_SATURATION = auto()