diff options
| -rw-r--r-- | simulation.py | 9 |
1 files changed, 5 insertions, 4 deletions
diff --git a/simulation.py b/simulation.py index ef05e4f..ba3945c 100644 --- a/simulation.py +++ b/simulation.py @@ -68,10 +68,11 @@ class Lattice: self.build_kernel() - if descriptor.d == 2: - self.layout = (32,1) - elif descriptor.d == 3: - self.layout = (32,1,1) + self.layout = { + (2, 9): (32,1), + (3,19): (32,4,4), + (3,27): (32,1,1) + }.get((descriptor.d, descriptor.q), None) self.program.equilibrilize( self.queue, self.geometry.span(), self.layout, self.cl_pop_a, self.cl_pop_b).wait() |