1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
|
{# ITERATE_ALL { _idx } #}
{# USES_VARIABLES { N } #}
{% extends 'common_group.py_' %}
{% block maincode %}
# scalar code
_vectorisation_idx = 1
{{scalar_code|autoindent}}
# vector code
_vectorisation_idx = LazyArange({{constant_or_scalar('N', variables['N'])}})
{{vector_code|autoindent}}
if _cond is True:
_cond = slice(None)
if _cond is False:
_cond = []
# _variable is set in the abstract code, see Group._get_with_code
_return_values = _variable[_cond]
{% endblock %}
|
