aboutsummaryrefslogtreecommitdiff
path: root/boltzgen/kernel/template/equilibrilize.cpp.mako
blob: 3b95a31d66fe2c8367e85fce017de26737324fb2 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
void equilibrilize(${float_type}* f_next,
                   ${float_type}* f_prev,
                   std::size_t gid)
{
    ${float_type}* preshifted_f_next = f_next + ${layout.cell_preshift('gid')};
    ${float_type}* preshifted_f_prev = f_prev + ${layout.cell_preshift('gid')};

% for i, w_i in enumerate(descriptor.w):
    preshifted_f_next[${layout.pop_offset(i)}] = ${w_i.evalf()};
    preshifted_f_prev[${layout.pop_offset(i)}] = ${w_i.evalf()};
% endfor
}