diff options
Diffstat (limited to 'simulation.py')
| -rw-r--r-- | simulation.py | 5 |
1 files changed, 4 insertions, 1 deletions
diff --git a/simulation.py b/simulation.py index 039486c..b64ff96 100644 --- a/simulation.py +++ b/simulation.py @@ -67,7 +67,10 @@ class Memory: self.float_type = float_type if align: - self.size_x = pad(grid.size_x, 32) + self.size_x = pad(grid.size_x, { + numpy.float32: 32, + numpy.float64: 16 + }.get(float_type, None)) else: self.size_x = grid.size_x |