File: group_variable_set_conditional.cpp

package info (click to toggle)
brian 2.9.0-2
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid
  • size: 6,872 kB
  • sloc: python: 51,820; cpp: 2,033; makefile: 108; sh: 72
file content (31 lines) | stat: -rw-r--r-- 979 bytes parent folder | download | duplicates (4)
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
{% extends 'common_group.cpp' %}
{# USES_VARIABLES { N } #}

{% block maincode %}
//// MAIN CODE ////////////
// scalar code
const size_t _vectorisation_idx = -1;
{# Note that the scalar_code['statement'] will not write to any scalar
   variables (except if the condition is simply 'True' and no vector code
   is present), it will only read in scalar variables that are used by the
   vector code. #}
{{scalar_code['condition']|autoindent}}
{{scalar_code['statement']|autoindent}}

{# N is a constant in most cases (NeuronGroup, etc.), but a scalar array for
   synapses, we therefore have to take care to get its value in the right
   way. #}
const int _N = {{constant_or_scalar('N', variables['N'])}};

{{ openmp_pragma('parallel-static') }}
for(int _idx=0; _idx<_N; _idx++)
{
    // vector code
    const size_t _vectorisation_idx = _idx;
    {{vector_code['condition']|autoindent}}
    if (_cond)
    {
        {{vector_code['statement']|autoindent}}
    }
}
{% endblock %}