diff options
Import `sympy.ccode` inside templates instead of as argument
Diffstat (limited to 'boltzgen/kernel/template/collect_moments.cl.mako')
-rw-r--r-- | boltzgen/kernel/template/collect_moments.cl.mako | 8 |
1 files changed, 6 insertions, 2 deletions
diff --git a/boltzgen/kernel/template/collect_moments.cl.mako b/boltzgen/kernel/template/collect_moments.cl.mako index 587f58a..39317e3 100644 --- a/boltzgen/kernel/template/collect_moments.cl.mako +++ b/boltzgen/kernel/template/collect_moments.cl.mako @@ -1,3 +1,7 @@ +<% +import sympy +%> + __kernel void collect_moments_gid(__global ${float_type}* f, __global ${float_type}* m, unsigned int gid) @@ -14,11 +18,11 @@ __kernel void collect_moments_gid(__global ${float_type}* f, %> % for i, expr in enumerate(moments_subexpr): - const ${float_type} ${expr[0]} = ${ccode(expr[1])}; + const ${float_type} ${expr[0]} = ${sympy.ccode(expr[1])}; % endfor % for i, expr in enumerate(moments_assignment): - preshifted_m[${i}] = ${ccode(expr.rhs)}; + preshifted_m[${i}] = ${sympy.ccode(expr.rhs)}; % endfor } |