aboutsummaryrefslogtreecommitdiff
path: root/boltzgen/kernel/template/collect_moments.cl.mako
diff options
context:
space:
mode:
authorAdrian Kummerlaender2019-11-02 17:29:56 +0100
committerAdrian Kummerlaender2019-11-02 17:29:56 +0100
commitf233beddfc74d5933f46684adab5298e03c08871 (patch)
tree5b9bd6151e5e6d411c2707c8cc5868c38124668f /boltzgen/kernel/template/collect_moments.cl.mako
parent24847cbb2567f508a7c30b39c6fb7ba6379d1adc (diff)
downloadboltzgen-f233beddfc74d5933f46684adab5298e03c08871.tar
boltzgen-f233beddfc74d5933f46684adab5298e03c08871.tar.gz
boltzgen-f233beddfc74d5933f46684adab5298e03c08871.tar.bz2
boltzgen-f233beddfc74d5933f46684adab5298e03c08871.tar.lz
boltzgen-f233beddfc74d5933f46684adab5298e03c08871.tar.xz
boltzgen-f233beddfc74d5933f46684adab5298e03c08871.tar.zst
boltzgen-f233beddfc74d5933f46684adab5298e03c08871.zip
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.mako8
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
}