diff options
| -rw-r--r-- | boltzgen/kernel/memory/layout/SOA.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/boltzgen/kernel/memory/layout/SOA.py b/boltzgen/kernel/memory/layout/SOA.py index 05e9e84..adb5200 100644 --- a/boltzgen/kernel/memory/layout/SOA.py +++ b/boltzgen/kernel/memory/layout/SOA.py @@ -7,8 +7,8 @@ class SOA: def cell_preshift(self, gid): return gid - def pop_offset(self, i): - return i * self.geometry.volume + def pop_offset(self, i, padding = 0): + return i * (self.geometry.volume + padding) def neighbor_offset(self, c_i): return self.index.neighbor(c_i) |