20 lines
814 B
C++
20 lines
814 B
C++
#pragma once
|
|
#include <knp/core/population.h>
|
|
#include <knp/core/projection.h>
|
|
#include <knp/core/messaging/messaging.h>
|
|
#include <knp/neuron-traits/all_traits.h>
|
|
#include <knp/synapse-traits/all_traits.h>
|
|
#include <functional>
|
|
#include <vector>
|
|
|
|
namespace knp_gpu_experiment
|
|
{
|
|
using Neuron = knp::neuron_traits::SynapticResourceSTDPAltAILIFNeuron;
|
|
using Synapse = knp::synapse_traits::SynapticResourceSTDPDeltaSynapse;
|
|
using Population = knp::core::Population<Neuron>;
|
|
using Projections = std::vector<std::reference_wrapper<knp::core::Projection<Synapse>>>;
|
|
using Reference = void (*)(Population&, Projections&, const knp::core::messaging::SpikeMessage&, knp::core::Step);
|
|
void train(Population&, Projections&, const knp::core::messaging::SpikeMessage&, knp::core::Step, Reference);
|
|
void flush();
|
|
void clear();
|
|
}
|