diff options
author | Adrian Kummerlaender | 2019-10-29 19:30:50 +0100 |
---|---|---|
committer | Adrian Kummerlaender | 2019-10-29 19:30:50 +0100 |
commit | a0438d434a8dde45e6fdab38d44181f0cd0cb2c6 (patch) | |
tree | 679d0262f69f3e1b3536cd66e3ef25a026c9a25e /boltzgen/kernel/memory | |
parent | a6dcb57cff9a1dd9de7e5fafdc87230489be87b9 (diff) | |
download | boltzgen-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__.py | 3 | ||||
-rw-r--r-- | boltzgen/kernel/memory/index.py | 18 | ||||
-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 | ||||
-rw-r--r-- | boltzgen/kernel/memory/precision.py | 16 |
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 |