diff options
Diffstat (limited to 'codegen_lbm.py')
-rw-r--r-- | codegen_lbm.py | 132 |
1 files changed, 0 insertions, 132 deletions
diff --git a/codegen_lbm.py b/codegen_lbm.py deleted file mode 100644 index a71fc15..0000000 --- a/codegen_lbm.py +++ /dev/null @@ -1,132 +0,0 @@ -import pyopencl as cl -mf = cl.mem_flags - -import numpy -import time - -import matplotlib -import matplotlib.pyplot as plt -matplotlib.use('AGG') - -import sympy -import lbm_d2q9 as D2Q9 - -from mako.template import Template - -class D2Q9_BGK_Lattice: - def idx(self, x, y): - return y * self.nX + x; - - def __init__(self, nX, nY, tau, geometry): - self.nX = nX - self.nY = nY - self.nCells = nX * nY - self.tau = tau - self.tick = True - - self.platform = cl.get_platforms()[0] - self.context = cl.Context(properties=[(cl.context_properties.PLATFORM, self.platform)]) - self.queue = cl.CommandQueue(self.context) - - self.np_material = numpy.ndarray(shape=(self.nCells, 1), dtype=numpy.int32) - self.setup_geometry(geometry) - - self.cl_pop_a = cl.Buffer(self.context, mf.READ_WRITE, size=9*self.nCells*numpy.float32(0).nbytes) - self.cl_pop_b = cl.Buffer(self.context, mf.READ_WRITE, size=9*self.nCells*numpy.float32(0).nbytes) - - self.cl_moments = cl.Buffer(self.context, mf.WRITE_ONLY, size=3*self.nCells*numpy.float32(0).nbytes) - self.cl_material = cl.Buffer(self.context, mf.READ_ONLY | mf.USE_HOST_PTR, hostbuf=self.np_material) - - self.build_kernel() - - self.program.equilibrilize(self.queue, (self.nX,self.nY), (32,1), self.cl_pop_a, self.cl_pop_b).wait() - - def setup_geometry(self, geometry): - for y in range(1,self.nY-1): - for x in range(1,self.nX-1): - self.np_material[self.idx(x,y)] = geometry(self.nX,self.nY,x,y) - - def build_kernel(self): - program_src = Template(filename = './template/kernel.mako').render( - nX = self.nX, - nY = self.nY, - nCells = self.nCells, - tau = self.tau, - moments_helper = D2Q9.moments_opt[0], - moments_assignment = D2Q9.moments_opt[1], - collide_helper = D2Q9.collide_opt[0], - collide_assignment = D2Q9.collide_opt[1], - c = D2Q9.c, - w = D2Q9.w, - ccode = sympy.ccode - ) - self.program = cl.Program(self.context, program_src).build() - - def evolve(self): - if self.tick: - self.tick = False - self.program.collide_and_stream(self.queue, (self.nX,self.nY), (32,1), self.cl_pop_a, self.cl_pop_b, self.cl_material) - else: - self.tick = True - self.program.collide_and_stream(self.queue, (self.nX,self.nY), (32,1), self.cl_pop_b, self.cl_pop_a, self.cl_material) - - def sync(self): - self.queue.finish() - - def get_moments(self): - moments = numpy.ndarray(shape=(3, self.nCells), dtype=numpy.float32) - if self.tick: - self.program.collect_moments(self.queue, (self.nX,self.nY), (32,1), self.cl_pop_b, self.cl_moments) - else: - self.program.collect_moments(self.queue, (self.nX,self.nY), (32,1), self.cl_pop_a, self.cl_moments) - cl.enqueue_copy(LBM.queue, moments, LBM.cl_moments).wait(); - return moments - - -def MLUPS(cells, steps, time): - return cells * steps / time * 1e-6 - -def generate_moment_plots(lattice, moments): - for i, m in enumerate(moments): - print("Generating plot %d of %d." % (i+1, len(moments))) - - density = numpy.ndarray(shape=(lattice.nY-2, lattice.nX-2)) - for y in range(1,lattice.nY-1): - for x in range(1,lattice.nX-1): - density[y-1,x-1] = m[0,lattice.idx(x,y)] - - plt.figure(figsize=(10, 10)) - plt.imshow(density, origin='lower', vmin=0.2, vmax=2.0, cmap=plt.get_cmap('seismic')) - plt.savefig("result/density_" + str(i) + ".png", bbox_inches='tight', pad_inches=0) - -def box(nX, nY, x, y): - if x == 1 or y == 1 or x == nX-2 or y == nY-2: - return 2 - else: - return 1 - -nUpdates = 1000 -nStat = 100 - -moments = [] - -print("Initializing simulation...\n") - -LBM = D2Q9_BGK_Lattice(nX = 1024, nY = 1024, tau = 0.8, geometry = box) - -print("Starting simulation using %d cells...\n" % LBM.nCells) - -lastStat = time.time() - -for i in range(1,nUpdates+1): - LBM.evolve() - - if i % nStat == 0: - LBM.sync() - print("i = %4d; %3.0f MLUPS" % (i, MLUPS(LBM.nCells, nStat, time.time() - lastStat))) - moments.append(LBM.get_moments()) - lastStat = time.time() - -print("\nConcluded simulation.\n") - -generate_moment_plots(LBM, moments) |