Added stdp altai support for single threaded cpu backend: KasperskyLab#64

Author: DavidIkov

knp/backends/cpu/cpu-library/include/knp/backends/cpu-library/impl/populations/altai/altai_dispatcher.h

@@ -41,6 +41,17 @@ inline void calculate_pre_impact_single_neuron_state_dispatch(
 }
 
 
+/**
+ * @brief Calculate pre impact state of single neuron.
+ * @param neuron Neuron.
+ */
+inline void calculate_pre_impact_single_neuron_state_dispatch(
+    knp::neuron_traits::neuron_parameters<knp::neuron_traits::SynapticResourceSTDPAltAILIFNeuron> &neuron)
+{
+    altai::calculate_pre_impact_single_neuron_state_impl(neuron);
+}
+
+
 /**
  * @brief Impact neuron.
  * @param neuron Neuron.

knp/backends/cpu/cpu-single-threaded-backend/impl/backend.cpp

@@ -241,6 +241,17 @@ std::optional<core::messaging::SpikeMessage> SingleThreadedCPUBackend::calculate
 }
 
 
+std::optional<core::messaging::SpikeMessage> SingleThreadedCPUBackend::calculate_population(
+    core::Population<neuron_traits::SynapticResourceSTDPAltAILIFNeuron> &population)
+{
+    SPDLOG_TRACE(
+        "Calculate resource-based STDP-compatible AltAI-LIF population {}.", std::string(population.get_uid()));
+    return knp::backends::cpu::calculate_resource_stdp_population<
+        neuron_traits::AltAILIF, synapse_traits::DeltaSynapse, ProjectionContainer>(
+        population, projections_, get_message_endpoint(), get_step());
+}
+
+
 void SingleThreadedCPUBackend::calculate_projection(
     knp::core::Projection<knp::synapse_traits::DeltaSynapse> &projection, SynapticMessageQueue &message_queue)
 {

knp/backends/cpu/cpu-single-threaded-backend/include/knp/backends/cpu-single-threaded/backend.h

@@ -57,7 +57,7 @@ class KNP_DECLSPEC SingleThreadedCPUBackend : public knp::core::Backend
      */
     using SupportedNeurons = boost::mp11::mp_list<
         knp::neuron_traits::BLIFATNeuron, knp::neuron_traits::SynapticResourceSTDPBLIFATNeuron,
-        knp::neuron_traits::AltAILIF>;
+        knp::neuron_traits::AltAILIF, knp::neuron_traits::SynapticResourceSTDPAltAILIFNeuron>;
 
     /**
      * @brief List of synapse types supported by the single-threaded CPU backend.
@@ -346,6 +346,17 @@ class KNP_DECLSPEC SingleThreadedCPUBackend : public knp::core::Backend
     std::optional<core::messaging::SpikeMessage> calculate_population(
         knp::core::Population<knp::neuron_traits::AltAILIF> &population);
 
+
+    /**
+     * @brief Calculate population of 'AltAILIF' neurons.
+     * @note Population state will be changed during calculation.
+     * @param population population to calculate.
+     * @return spike message with indexes of spiked neurons if population is emitting one.
+     */
+    std::optional<core::messaging::SpikeMessage> calculate_population(
+        knp::core::Population<knp::neuron_traits::SynapticResourceSTDPAltAILIFNeuron> &population);
+
+
     /**
      * @brief Calculate projection of delta synapses.
      * @note Projection will be changed during calculation.