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.cl.mako | 4 ++++
 1 file changed, 4 insertions(+)

(limited to 'boltzgen/kernel/template/collect_moments.cl.mako')

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
-- 
cgit v1.2.3