Move OpenCL buffers into Memory class

1 files changed, 1 insertions, 1 deletions

diff --git a/implosion.py b/implosion.py
index 370ef92..245f2bb 100644
--- a/implosion.py
+++ b/implosion.py
@@ -19,7 +19,7 @@ def generate_moment_plots(lattice, moments):
 
         velocity = numpy.ndarray(shape=tuple(reversed(lattice.geometry.inner_size())))
         for x, y in lattice.geometry.inner_cells():
-            velocity[y-1,x-1] = numpy.sqrt(m[1,lattice.gid(x,y)]**2 + m[2,lattice.gid(x,y)]**2)
+            velocity[y-1,x-1] = numpy.sqrt(m[1,lattice.memory.gid(x,y)]**2 + m[2,lattice.memory.gid(x,y)]**2)
 
         plt.figure(figsize=(10, 10))
         plt.imshow(velocity, origin='lower', cmap=plt.get_cmap('seismic'))