From f233beddfc74d5933f46684adab5298e03c08871 Mon Sep 17 00:00:00 2001
From: Adrian Kummerlaender
Date: Sat, 2 Nov 2019 17:29:56 +0100
Subject: Import `sympy.ccode` inside templates instead of as argument

---
 boltzgen/kernel/template/collide_and_stream.cpp.mako | 9 ++++++---
 1 file changed, 6 insertions(+), 3 deletions(-)

(limited to 'boltzgen/kernel/template/collide_and_stream.cpp.mako')

diff --git a/boltzgen/kernel/template/collide_and_stream.cpp.mako b/boltzgen/kernel/template/collide_and_stream.cpp.mako
index 2c52b54..ee42eb7 100644
--- a/boltzgen/kernel/template/collide_and_stream.cpp.mako
+++ b/boltzgen/kernel/template/collide_and_stream.cpp.mako
@@ -1,3 +1,7 @@
+<%
+import sympy
+%>
+
 void collide_and_stream(      ${float_type}* f_next,
                         const ${float_type}* f_prev,
                         std::size_t gid)
@@ -14,15 +18,14 @@ void collide_and_stream(      ${float_type}* f_next,
 %>
 
 % for i, expr in enumerate(subexpr):
-    const ${float_type} ${expr[0]} = ${ccode(expr[1])};
+    const ${float_type} ${expr[0]} = ${sympy.ccode(expr[1])};
 % endfor
 
 % for i, expr in enumerate(assignment):
-    const ${float_type} ${ccode(expr)}
+    const ${float_type} ${sympy.ccode(expr)}
 % endfor
 
 % for i, expr in enumerate(assignment):
     preshifted_f_next[${layout.pop_offset(i)}] = f_next_${i};
 % endfor
 }
-
-- 
cgit v1.2.3