aboutsummaryrefslogtreecommitdiff
path: root/boltzgen/kernel/memory
diff options
context:
space:
mode:
authorAdrian Kummerlaender2019-10-29 19:30:50 +0100
committerAdrian Kummerlaender2019-10-29 19:30:50 +0100
commita0438d434a8dde45e6fdab38d44181f0cd0cb2c6 (patch)
tree679d0262f69f3e1b3536cd66e3ef25a026c9a25e /boltzgen/kernel/memory
parenta6dcb57cff9a1dd9de7e5fafdc87230489be87b9 (diff)
downloadboltzgen-a0438d434a8dde45e6fdab38d44181f0cd0cb2c6.tar
boltzgen-a0438d434a8dde45e6fdab38d44181f0cd0cb2c6.tar.gz
boltzgen-a0438d434a8dde45e6fdab38d44181f0cd0cb2c6.tar.bz2
boltzgen-a0438d434a8dde45e6fdab38d44181f0cd0cb2c6.tar.lz
boltzgen-a0438d434a8dde45e6fdab38d44181f0cd0cb2c6.tar.xz
boltzgen-a0438d434a8dde45e6fdab38d44181f0cd0cb2c6.tar.zst
boltzgen-a0438d434a8dde45e6fdab38d44181f0cd0cb2c6.zip
Rename target module to memory
Diffstat (limited to 'boltzgen/kernel/memory')
-rw-r--r--boltzgen/kernel/memory/__init__.py3
-rw-r--r--boltzgen/kernel/memory/index.py18
-rw-r--r--boltzgen/kernel/memory/layout/AOS.py14
-rw-r--r--boltzgen/kernel/memory/layout/SOA.py14
-rw-r--r--boltzgen/kernel/memory/layout/__init__.py2
-rw-r--r--boltzgen/kernel/memory/precision.py16
6 files changed, 67 insertions, 0 deletions
diff --git a/boltzgen/kernel/memory/__init__.py b/boltzgen/kernel/memory/__init__.py
new file mode 100644
index 0000000..5927d1b
--- /dev/null
+++ b/boltzgen/kernel/memory/__init__.py
@@ -0,0 +1,3 @@
+from . import layout
+from . import precision
+from . import index
diff --git a/boltzgen/kernel/memory/index.py b/boltzgen/kernel/memory/index.py
new file mode 100644
index 0000000..e89e9ee
--- /dev/null
+++ b/boltzgen/kernel/memory/index.py
@@ -0,0 +1,18 @@
+class CellIndex:
+ def __init__(self, geometry):
+ self.geometry = geometry
+
+ def neighbor(self, c_i):
+ return self.gid(*[ c for _, c in enumerate(c_i) ])
+
+class XYZ(CellIndex):
+ pass
+
+ def gid(self, x, y, z=0):
+ return x*self.geometry.size_y*self.geometry.size_z + y*self.geometry.size_z + z
+
+class ZYX(CellIndex):
+ pass
+
+ def gid(self, x, y, z=0):
+ return z*self.geometry.size_x*self.geometry.size_y + y*self.geometry.size_x + x
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
diff --git a/boltzgen/kernel/memory/precision.py b/boltzgen/kernel/memory/precision.py
new file mode 100644
index 0000000..d22e922
--- /dev/null
+++ b/boltzgen/kernel/memory/precision.py
@@ -0,0 +1,16 @@
+class common:
+ @classmethod
+ def get_float_type(self, precision):
+ if precision not in ['single', 'double']:
+ raise Exception("Precision must be either 'single' or 'double'")
+
+ return {
+ 'single': 'float',
+ 'double': 'double'
+ }.get(precision)
+
+class cpp(common):
+ pass
+
+class cl(common):
+ pass