1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
|
{% macro cpp_file() %}
{% endmacro %}
{% macro h_file() %}
#ifndef _BRIAN_SYNAPSES_H
#define _BRIAN_SYNAPSES_H
#include<vector>
#include<algorithm>
{{ openmp_pragma('include') }}
#include "brianlib/spikequeue.h"
class SynapticPathway
{
public:
int Nsource, Ntarget, _nb_threads;
std::vector<int> &sources;
std::vector<int> all_peek;
std::vector< CSpikeQueue * > queue;
SynapticPathway(std::vector<int> &_sources, int _spikes_start, int _spikes_stop)
: sources(_sources)
{
_nb_threads = {{ openmp_pragma('get_num_threads') }};
for (int _idx=0; _idx < _nb_threads; _idx++)
queue.push_back(new CSpikeQueue(_spikes_start, _spikes_stop));
};
~SynapticPathway()
{
for (int _idx=0; _idx < _nb_threads; _idx++)
delete(queue[_idx]);
}
void push(int *spikes, int nspikes)
{
queue[{{ openmp_pragma('get_thread_num') }}]->push(spikes, nspikes);
}
void advance()
{
queue[{{ openmp_pragma('get_thread_num') }}]->advance();
}
vector<int32_t>* peek()
{
{{ openmp_pragma('static-ordered') }}
for(int _thread=0; _thread < {{ openmp_pragma('get_num_threads') }}; _thread++)
{
{{ openmp_pragma('ordered') }}
{
if (_thread == 0)
all_peek.clear();
all_peek.insert(all_peek.end(), queue[_thread]->peek()->begin(), queue[_thread]->peek()->end());
}
}
return &all_peek;
}
template <typename scalar> void prepare(int n_source, int n_target, scalar *real_delays, int n_delays,
int *sources, int n_synapses, double _dt)
{
Nsource = n_source;
Ntarget = n_target;
{{ openmp_pragma('parallel') }}
{
int length;
if ({{ openmp_pragma('get_thread_num') }} == _nb_threads - 1)
length = n_synapses - (int){{ openmp_pragma('get_thread_num') }}*(n_synapses/_nb_threads);
else
length = (int) n_synapses/_nb_threads;
int padding = {{ openmp_pragma('get_thread_num') }}*(n_synapses/_nb_threads);
queue[{{ openmp_pragma('get_thread_num') }}]->openmp_padding = padding;
if (n_delays > 1)
queue[{{ openmp_pragma('get_thread_num') }}]->prepare(&real_delays[padding], length, &sources[padding], length, _dt);
else if (n_delays == 1)
queue[{{ openmp_pragma('get_thread_num') }}]->prepare(&real_delays[0], 1, &sources[padding], length, _dt);
else // no synapses
queue[{{ openmp_pragma('get_thread_num') }}]->prepare((scalar *)NULL, 0, &sources[padding], length, _dt);
}
}
};
#endif
{% endmacro %}
|
