aboutsummaryrefslogtreecommitdiff
path: root/boltzgen/kernel/template/equilibrilize.cl.mako
diff options
context:
space:
mode:
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>