aboutsummaryrefslogtreecommitdiff
path: root/boltzgen/kernel/template/momenta_boundary.cl.mako
diff options
context:
space:
mode:
Diffstat (limited to 'boltzgen/kernel/template/momenta_boundary.cl.mako')
-rw-r--r--boltzgen/kernel/template/momenta_boundary.cl.mako5
1 files changed, 5 insertions, 0 deletions
diff --git a/boltzgen/kernel/template/momenta_boundary.cl.mako b/boltzgen/kernel/template/momenta_boundary.cl.mako
index f5edb61..4e29601 100644
--- a/boltzgen/kernel/template/momenta_boundary.cl.mako
+++ b/boltzgen/kernel/template/momenta_boundary.cl.mako
@@ -1,3 +1,8 @@
+<%
+moments_subexpr, moments_assignment = model.moments()
+collision_subexpr, collision_assignment = model.bgk(f_eq = model.equilibrium(resolve_moments = False))
+%>
+
<%def name="momenta_boundary(name, param)">
__kernel void ${name}_momenta_boundary_gid(
__global ${float_type}* f_next,