diff options
author | Adrian Kummerlaender | 2019-10-21 22:23:41 +0200 |
---|---|---|
committer | Adrian Kummerlaender | 2019-10-21 22:24:51 +0200 |
commit | 8f88aea2c963ebe63db83480fc9a4b48451220af (patch) | |
tree | 0213cb1a46fc38e3fcf6dc2d3d6337beb06e3818 /boltzgen/kernel/template | |
parent | 82a44e0d64afb8818ea98d68dc08108885d503c2 (diff) | |
download | boltzgen-8f88aea2c963ebe63db83480fc9a4b48451220af.tar boltzgen-8f88aea2c963ebe63db83480fc9a4b48451220af.tar.gz boltzgen-8f88aea2c963ebe63db83480fc9a4b48451220af.tar.bz2 boltzgen-8f88aea2c963ebe63db83480fc9a4b48451220af.tar.lz boltzgen-8f88aea2c963ebe63db83480fc9a4b48451220af.tar.xz boltzgen-8f88aea2c963ebe63db83480fc9a4b48451220af.tar.zst boltzgen-8f88aea2c963ebe63db83480fc9a4b48451220af.zip |
Pull in C++ template from symlbm_playground's standalone branch
Diffstat (limited to 'boltzgen/kernel/template')
-rw-r--r-- | boltzgen/kernel/template/basic.cpp.mako | 66 | ||||
-rw-r--r-- | boltzgen/kernel/template/basic.opencl.mako (renamed from boltzgen/kernel/template/kernel.mako) | 0 |
2 files changed, 66 insertions, 0 deletions
diff --git a/boltzgen/kernel/template/basic.cpp.mako b/boltzgen/kernel/template/basic.cpp.mako new file mode 100644 index 0000000..1879c35 --- /dev/null +++ b/boltzgen/kernel/template/basic.cpp.mako @@ -0,0 +1,66 @@ +<% +def pop_offset(i): + return i * geometry.volume +%> + +void equilibrilize(${float_type}* f_next, + ${float_type}* f_prev, + const std::size_t gid) +{ + ${float_type}* preshifted_f_next = f_next + gid; + ${float_type}* preshifted_f_prev = f_prev + gid; + +% for i, w_i in enumerate(descriptor.w): + preshifted_f_next[${pop_offset(i)}] = ${w_i.evalf()}; + preshifted_f_prev[${pop_offset(i)}] = ${w_i.evalf()}; +% endfor +} + +<% +def neighbor_offset(c_i): + return { + 2: lambda: c_i[1]*geometry.size_x + c_i[0], + 3: lambda: c_i[2]*geometry.size_x*geometry.size_y + c_i[1]*geometry.size_x + c_i[0] + }.get(descriptor.d)() + +def padding(): + return { + 2: lambda: 1*geometry.size_x + 1, + 3: lambda: 1*geometry.size_x*geometry.size_y + 1*geometry.size_x + 1 + }.get(descriptor.d)() +%> + +void collide_and_stream( ${float_type}* f_next, + const ${float_type}* f_prev, + const int* material, + const std::size_t gid) +{ + const int m = material[gid]; + + ${float_type}* preshifted_f_next = f_next + gid; + const ${float_type}* preshifted_f_prev = f_prev + gid; + +% for i, c_i in enumerate(descriptor.c): + const ${float_type} f_curr_${i} = preshifted_f_prev[${pop_offset(i) + neighbor_offset(-c_i)}]; +% endfor + +% for i, expr in enumerate(moments_subexpr): + const ${float_type} ${expr[0]} = ${ccode(expr[1])}; +% endfor + +% for i, expr in enumerate(moments_assignment): + ${float_type} ${ccode(expr)} +% endfor + +% for i, expr in enumerate(collide_subexpr): + const ${float_type} ${expr[0]} = ${ccode(expr[1])}; +% endfor + +% for i, expr in enumerate(collide_assignment): + const ${float_type} ${ccode(expr)} +% endfor + +% for i, expr in enumerate(collide_assignment): + preshifted_f_next[${pop_offset(i)}] = m*f_next_${i} + (1.0-m)*${descriptor.w[i].evalf()}; +% endfor +} diff --git a/boltzgen/kernel/template/kernel.mako b/boltzgen/kernel/template/basic.opencl.mako index 5ddf64c..5ddf64c 100644 --- a/boltzgen/kernel/template/kernel.mako +++ b/boltzgen/kernel/template/basic.opencl.mako |