From e2b00f4ec963060be98939c7b0d12d6c00e50a02 Mon Sep 17 00:00:00 2001 From: Adrian Kummerlaender Date: Thu, 31 Oct 2019 13:13:00 +0100 Subject: Call symbolic generator inside code templates This paves the way for dropping in other LBM collision models. As a side benefit the default momenta calulcation is now fully inlined where possible. --- boltzgen/kernel/template/collect_moments.cpp.mako | 4 ++++ 1 file changed, 4 insertions(+) (limited to 'boltzgen/kernel/template/collect_moments.cpp.mako') diff --git a/boltzgen/kernel/template/collect_moments.cpp.mako b/boltzgen/kernel/template/collect_moments.cpp.mako index d72ff5d..1605084 100644 --- a/boltzgen/kernel/template/collect_moments.cpp.mako +++ b/boltzgen/kernel/template/collect_moments.cpp.mako @@ -9,6 +9,10 @@ void collect_moments(const ${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 -- cgit v1.2.3