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/bounce_back_boundary.cpp.mako | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) (limited to 'boltzgen/kernel/template/bounce_back_boundary.cpp.mako') diff --git a/boltzgen/kernel/template/bounce_back_boundary.cpp.mako b/boltzgen/kernel/template/bounce_back_boundary.cpp.mako index c37233e..1dcafe8 100644 --- a/boltzgen/kernel/template/bounce_back_boundary.cpp.mako +++ b/boltzgen/kernel/template/bounce_back_boundary.cpp.mako @@ -1,3 +1,7 @@ +<% +import sympy +%> + void bounce_back_boundary( ${float_type}* f_next, const ${float_type}* f_prev, std::size_t gid) @@ -14,11 +18,11 @@ void bounce_back_boundary( ${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, c_i in enumerate(descriptor.c): -- cgit v1.2.3