aboutsummaryrefslogtreecommitdiff
path: root/simulation.py
diff options
context:
space:
mode:
authorAdrian Kummerlaender2019-10-29 20:41:33 +0100
committerAdrian Kummerlaender2019-10-29 20:41:33 +0100
commit39aa2832b3d8e1a3e1dd2efb0bf9629d8190ed39 (patch)
treee2fce29ab9e5fa24408352b7c7155cc464e7200b /simulation.py
parent12e8d10df34e8def9a5fb3944bff6de25f020d27 (diff)
downloadboltzgen_examples-39aa2832b3d8e1a3e1dd2efb0bf9629d8190ed39.tar
boltzgen_examples-39aa2832b3d8e1a3e1dd2efb0bf9629d8190ed39.tar.gz
boltzgen_examples-39aa2832b3d8e1a3e1dd2efb0bf9629d8190ed39.tar.bz2
boltzgen_examples-39aa2832b3d8e1a3e1dd2efb0bf9629d8190ed39.tar.lz
boltzgen_examples-39aa2832b3d8e1a3e1dd2efb0bf9629d8190ed39.tar.xz
boltzgen_examples-39aa2832b3d8e1a3e1dd2efb0bf9629d8190ed39.tar.zst
boltzgen_examples-39aa2832b3d8e1a3e1dd2efb0bf9629d8190ed39.zip
Adapt to upstream changes
Diffstat (limited to 'simulation.py')
-rw-r--r--simulation.py3
1 files changed, 2 insertions, 1 deletions
diff --git a/simulation.py b/simulation.py
index e0ce3ba..7625609 100644
--- a/simulation.py
+++ b/simulation.py
@@ -41,6 +41,7 @@ class CellList:
class Lattice:
def __init__(self, geometry, kernel_src, descriptor, platform = 0, precision = 'single'):
self.geometry = geometry
+ self.descriptor = descriptor
self.float_type = {
'single': (numpy.float32, 'float'),
@@ -90,7 +91,7 @@ class Lattice:
self.queue.finish()
def get_moments(self):
- moments = numpy.ndarray(shape=(3, self.memory.volume), dtype=self.float_type[0])
+ moments = numpy.ndarray(shape=(self.memory.volume*(self.descriptor.d+1),1), dtype=self.float_type[0])
if self.tick:
self.program.collect_moments(