diff options
Diffstat (limited to 'boltzgen/kernel/memory/layout')
-rw-r--r-- | boltzgen/kernel/memory/layout/AOS.py | 14 | ||||
-rw-r--r-- | boltzgen/kernel/memory/layout/SOA.py | 14 | ||||
-rw-r--r-- | boltzgen/kernel/memory/layout/__init__.py | 2 |
3 files changed, 30 insertions, 0 deletions
diff --git a/boltzgen/kernel/memory/layout/AOS.py b/boltzgen/kernel/memory/layout/AOS.py new file mode 100644 index 0000000..88e5377 --- /dev/null +++ b/boltzgen/kernel/memory/layout/AOS.py @@ -0,0 +1,14 @@ +class AOS: + def __init__(self, descriptor, index, geometry): + self.descriptor = descriptor + self.index = index(geometry) + self.geometry = geometry + + def cell_preshift(self, gid): + return "(%s)*%d" % (gid, self.descriptor.q) + + def pop_offset(self, i): + return i + + def neighbor_offset(self, c_i): + return self.descriptor.q * self.index.neighbor(c_i) diff --git a/boltzgen/kernel/memory/layout/SOA.py b/boltzgen/kernel/memory/layout/SOA.py new file mode 100644 index 0000000..05e9e84 --- /dev/null +++ b/boltzgen/kernel/memory/layout/SOA.py @@ -0,0 +1,14 @@ +class SOA: + def __init__(self, descriptor, index, geometry): + self.descriptor = descriptor + self.index = index(geometry) + self.geometry = geometry + + def cell_preshift(self, gid): + return gid + + def pop_offset(self, i): + return i * self.geometry.volume + + def neighbor_offset(self, c_i): + return self.index.neighbor(c_i) diff --git a/boltzgen/kernel/memory/layout/__init__.py b/boltzgen/kernel/memory/layout/__init__.py new file mode 100644 index 0000000..731fbe8 --- /dev/null +++ b/boltzgen/kernel/memory/layout/__init__.py @@ -0,0 +1,2 @@ +from .AOS import AOS +from .SOA import SOA |