diff options
Diffstat (limited to 'boltzgen/kernel/template/collide.cl.mako')
| -rw-r--r-- | boltzgen/kernel/template/collide.cl.mako | 15 |
1 files changed, 15 insertions, 0 deletions
diff --git a/boltzgen/kernel/template/collide.cl.mako b/boltzgen/kernel/template/collide.cl.mako new file mode 100644 index 0000000..51b591a --- /dev/null +++ b/boltzgen/kernel/template/collide.cl.mako @@ -0,0 +1,15 @@ +<%namespace name="pattern" file="${'/pattern/%s.cl.mako' % context['streaming']}"/> +<% +import sympy +subexpr, assignment = model.collision(f_eq = model.equilibrium(resolve_moments = True)) +%> + +<%call expr="pattern.operator('collide')"> +% for i, expr in enumerate(subexpr): + const ${float_type} ${expr[0]} = ${sympy.ccode(expr[1])}; +% endfor + +% for i, expr in enumerate(assignment): + const ${float_type} ${sympy.ccode(expr)} +% endfor +</%call> |