diff options
Diffstat (limited to 'boltzgen/kernel/template/collect_moments.cl.mako')
| -rw-r--r-- | boltzgen/kernel/template/collect_moments.cl.mako | 4 |
1 files changed, 4 insertions, 0 deletions
diff --git a/boltzgen/kernel/template/collect_moments.cl.mako b/boltzgen/kernel/template/collect_moments.cl.mako index ddf5102..587f58a 100644 --- a/boltzgen/kernel/template/collect_moments.cl.mako +++ b/boltzgen/kernel/template/collect_moments.cl.mako @@ -9,6 +9,10 @@ __kernel void collect_moments_gid(__global ${float_type}* f, const ${float_type} f_curr_${i} = preshifted_f[${layout.pop_offset(i)}]; % endfor +<% + moments_subexpr, moments_assignment = model.moments() +%> + % for i, expr in enumerate(moments_subexpr): const ${float_type} ${expr[0]} = ${ccode(expr[1])}; % endfor |