From dbd9a340a7809a770d52d10154712278431acdc3 Mon Sep 17 00:00:00 2001
From: Adrian Kummerlaender
Date: Tue, 29 Oct 2019 10:56:41 +0100
Subject: Unify AOS, SOA specific cell preshift between targets

SOA and AOS should not be target specific, neighbor offset calculation /
bijection between gid and cell coordinates should be customizable.
---
 boltzgen/kernel/template/equilibrilize.cpp.mako | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

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

diff --git a/boltzgen/kernel/template/equilibrilize.cpp.mako b/boltzgen/kernel/template/equilibrilize.cpp.mako
index 9f082a1..3b95a31 100644
--- a/boltzgen/kernel/template/equilibrilize.cpp.mako
+++ b/boltzgen/kernel/template/equilibrilize.cpp.mako
@@ -2,8 +2,8 @@ void equilibrilize(${float_type}* f_next,
                    ${float_type}* f_prev,
                    std::size_t gid)
 {
-    ${float_type}* preshifted_f_next = f_next + gid*${layout.gid_offset()};
-    ${float_type}* preshifted_f_prev = f_prev + gid*${layout.gid_offset()};
+    ${float_type}* preshifted_f_next = f_next + ${layout.cell_preshift('gid')};
+    ${float_type}* preshifted_f_prev = f_prev + ${layout.cell_preshift('gid')};
 
 % for i, w_i in enumerate(descriptor.w):
     preshifted_f_next[${layout.pop_offset(i)}] = ${w_i.evalf()};
-- 
cgit v1.2.3