aboutsummaryrefslogtreecommitdiff
path: root/boltzgen/kernel/template/equilibrilize.cl.mako
diff options
context:
space:
mode:
authorAdrian Kummerlaender2019-11-12 22:54:11 +0100
committerAdrian Kummerlaender2019-11-12 22:54:11 +0100
commitaa509dd4ebbb9d1d8ad6ebfe05111228fd9ae7c0 (patch)
treee43112bf6a79f2d1a92754391f01110fd335eeef /boltzgen/kernel/template/equilibrilize.cl.mako
parenta93535c32231d98ef8d080adac626f88b18f9db5 (diff)
downloadboltzgen-aa509dd4ebbb9d1d8ad6ebfe05111228fd9ae7c0.tar
boltzgen-aa509dd4ebbb9d1d8ad6ebfe05111228fd9ae7c0.tar.gz
boltzgen-aa509dd4ebbb9d1d8ad6ebfe05111228fd9ae7c0.tar.bz2
boltzgen-aa509dd4ebbb9d1d8ad6ebfe05111228fd9ae7c0.tar.lz
boltzgen-aa509dd4ebbb9d1d8ad6ebfe05111228fd9ae7c0.tar.xz
boltzgen-aa509dd4ebbb9d1d8ad6ebfe05111228fd9ae7c0.tar.zst
boltzgen-aa509dd4ebbb9d1d8ad6ebfe05111228fd9ae7c0.zip
Match OpenCL and CUDA cell list dispatch templates
Diffstat (limited to 'boltzgen/kernel/template/equilibrilize.cl.mako')
-rw-r--r--boltzgen/kernel/template/equilibrilize.cl.mako6
1 files changed, 6 insertions, 0 deletions
diff --git a/boltzgen/kernel/template/equilibrilize.cl.mako b/boltzgen/kernel/template/equilibrilize.cl.mako
index 8a423ae..36ec48b 100644
--- a/boltzgen/kernel/template/equilibrilize.cl.mako
+++ b/boltzgen/kernel/template/equilibrilize.cl.mako
@@ -5,3 +5,9 @@
const ${float_type} f_next_${i} = ${w_i.evalf()};
% endfor
</%call>
+
+<%call expr="pattern.operator_with_domain_dispatch('equilibrilize_domain')">
+% for i, w_i in enumerate(descriptor.w):
+ const ${float_type} f_next_${i} = ${w_i.evalf()};
+% endfor
+</%call>