From 621dec0f2b7a982c958cce4d7baa4ccd3994606a Mon Sep 17 00:00:00 2001
From: Adrian Kummerlaender
Date: Fri, 19 Jun 2020 21:30:34 +0200
Subject: Use OpenCL buffer to access moments in streamline impl

---
 simulation.py | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

(limited to 'simulation.py')

diff --git a/simulation.py b/simulation.py
index 5ebe9f6..e2174ee 100644
--- a/simulation.py
+++ b/simulation.py
@@ -261,9 +261,7 @@ class Lattice:
     def sync(self):
         self.queue.finish()
 
-    def get_moments(self):
-        moments = numpy.ndarray(shape=(self.descriptor.d+1, self.memory.volume), dtype=self.float_type[0])
-
+    def update_moments(self):
         if self.tick:
             self.program.collect_moments(
                 self.queue, self.grid.size(), self.layout, self.memory.cl_pop_b, self.memory.cl_moments)
@@ -271,6 +269,8 @@ class Lattice:
             self.program.collect_moments(
                 self.queue, self.grid.size(), self.layout, self.memory.cl_pop_a, self.memory.cl_moments)
 
+    def get_moments(self):
+        moments = numpy.ndarray(shape=(self.descriptor.d+1, self.memory.volume), dtype=self.float_type[0])
+        self.update_moments()
         cl.enqueue_copy(self.queue, moments, self.memory.cl_moments).wait();
-
         return moments
-- 
cgit v1.2.3