From aa509dd4ebbb9d1d8ad6ebfe05111228fd9ae7c0 Mon Sep 17 00:00:00 2001
From: Adrian Kummerlaender
Date: Tue, 12 Nov 2019 22:54:11 +0100
Subject: Match OpenCL and CUDA cell list dispatch templates

---
 boltzgen/kernel/template/equilibrilize.cl.mako | 6 ++++++
 1 file changed, 6 insertions(+)

(limited to 'boltzgen/kernel/template/equilibrilize.cl.mako')

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>
-- 
cgit v1.2.3