aboutsummaryrefslogtreecommitdiff
path: root/simulation.py
diff options
context:
space:
mode:
authorAdrian Kummerlaender2019-10-29 20:46:21 +0100
committerAdrian Kummerlaender2019-10-29 20:50:49 +0100
commit6bf667eeb0c8f40e0f6a28df1ecf4a40824771c3 (patch)
tree4a9b08b021471f4ff0009785ea5a5ed92443e066 /simulation.py
parent39aa2832b3d8e1a3e1dd2efb0bf9629d8190ed39 (diff)
downloadboltzgen_examples-6bf667eeb0c8f40e0f6a28df1ecf4a40824771c3.tar
boltzgen_examples-6bf667eeb0c8f40e0f6a28df1ecf4a40824771c3.tar.gz
boltzgen_examples-6bf667eeb0c8f40e0f6a28df1ecf4a40824771c3.tar.bz2
boltzgen_examples-6bf667eeb0c8f40e0f6a28df1ecf4a40824771c3.tar.lz
boltzgen_examples-6bf667eeb0c8f40e0f6a28df1ecf4a40824771c3.tar.xz
boltzgen_examples-6bf667eeb0c8f40e0f6a28df1ecf4a40824771c3.tar.zst
boltzgen_examples-6bf667eeb0c8f40e0f6a28df1ecf4a40824771c3.zip
Move example into subfolder
Diffstat (limited to 'simulation.py')
-rw-r--r--simulation.py105
1 files changed, 0 insertions, 105 deletions
diff --git a/simulation.py b/simulation.py
deleted file mode 100644
index 7625609..0000000
--- a/simulation.py
+++ /dev/null
@@ -1,105 +0,0 @@
-import pyopencl as cl
-mf = cl.mem_flags
-
-import numpy
-
-class Memory:
- def __init__(self, descriptor, geometry, context, float_type):
- self.context = context
- self.float_type = float_type
-
- self.size_x = geometry.size_x
- self.size_y = geometry.size_y
- self.size_z = geometry.size_z
- self.volume = self.size_x * self.size_y * self.size_z
-
- self.pop_size = descriptor.q * self.volume * self.float_type(0).nbytes
- self.moments_size = 3 * self.volume * self.float_type(0).nbytes
-
- self.cl_pop_a = cl.Buffer(self.context, mf.READ_WRITE, size=self.pop_size)
- self.cl_pop_b = cl.Buffer(self.context, mf.READ_WRITE, size=self.pop_size)
-
- self.cl_moments = cl.Buffer(self.context, mf.WRITE_ONLY, size=self.moments_size)
-
- def gid(self, x, y, z = 0):
- return z * (self.size_x*self.size_y) + y * self.size_x + x;
-
-class CellList:
- def __init__(self, context, queue, float_type, cells):
- self.cl_cells = cl.Buffer(context, mf.READ_ONLY, size=len(cells) * numpy.uint32(0).nbytes)
- self.np_cells = numpy.ndarray(shape=(len(cells), 1), dtype=numpy.uint32)
- self.np_cells[:,0] = cells[:]
-
- cl.enqueue_copy(queue, self.cl_cells, self.np_cells).wait();
-
- def get(self):
- return self.cl_cells
-
- def size(self):
- return (len(self.np_cells), 1, 1)
-
-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'),
- 'double': (numpy.float64, 'double'),
- }.get(precision, None)
-
- self.platform = cl.get_platforms()[platform]
- self.layout = None
-
- self.context = cl.Context(
- properties=[(cl.context_properties.PLATFORM, self.platform)])
-
- self.queue = cl.CommandQueue(self.context)
-
- self.memory = Memory(descriptor, self.geometry, self.context, self.float_type[0])
- self.tick = False
-
- self.compiler_args = {
- 'single': '-cl-single-precision-constant -cl-fast-relaxed-math',
- 'double': '-cl-fast-relaxed-math'
- }.get(precision, None)
-
- self.build_kernel(kernel_src)
-
- self.program.equilibrilize(
- self.queue, self.geometry.size(), self.layout, self.memory.cl_pop_a, self.memory.cl_pop_b).wait()
-
- self.tasks = []
-
- def build_kernel(self, src):
- self.program = cl.Program(self.context, src).build(self.compiler_args)
-
- def schedule(self, f, cells, *params):
- self.tasks += [ (eval("self.program.%s" % f), cells, params) ]
-
- def evolve(self):
- if self.tick:
- self.tick = False
- for f, cells, params in self.tasks:
- f(self.queue, cells.size(), self.layout, self.memory.cl_pop_a, self.memory.cl_pop_b, cells.get(), *params)
- else:
- self.tick = True
- for f, cells, params in self.tasks:
- f(self.queue, cells.size(), self.layout, self.memory.cl_pop_b, self.memory.cl_pop_a, cells.get(), *params)
-
- def sync(self):
- self.queue.finish()
-
- def get_moments(self):
- moments = numpy.ndarray(shape=(self.memory.volume*(self.descriptor.d+1),1), dtype=self.float_type[0])
-
- if self.tick:
- self.program.collect_moments(
- self.queue, self.geometry.size(), self.layout, self.memory.cl_pop_b, self.memory.cl_moments)
- else:
- self.program.collect_moments(
- self.queue, self.geometry.size(), self.layout, self.memory.cl_pop_a, self.memory.cl_moments)
-
- cl.enqueue_copy(self.queue, moments, self.memory.cl_moments).wait();
-
- return moments