1 files changed, 8 insertions, 7 deletions
diff --git a/sim.py b/sim.py
index 26835f9..7ba6472 100755
--- a/sim.py
+++ b/sim.py
@@ -48,10 +48,7 @@ class Simulation(object):
             detector.build(bits=bvh_bits, use_cache=use_cache)
         
         self.gpu = gpu.GPU(cuda_device)
-
-        # geometry is loaded onto gpu by default
-        self.gpu_geometry = gpu.GPUGeometry(self.gpu, detector)
-        
+        self.gpu_geometry = gpu.GPUGeometry(detector)
         self.gpu_daq = gpu.GPUDaq(self.gpu_geometry, max(self.detector.pmtids))
         self.gpu_pdf = gpu.GPUPDF()
 
@@ -77,7 +74,7 @@ class Simulation(object):
         for ev in iterable:
             gpu_photons = gpu.GPUPhotons(ev.photons_beg)
 
-            gpu.propagate(self.gpu, gpu_photons, self.rng_states, nthreads_per_block=self.nthreads_per_block, max_blocks=self.max_blocks, max_steps=max_steps)
+            gpu_photons.propagate(self.gpu_geometry, self.rng_states, nthreads_per_block=self.nthreads_per_block, max_blocks=self.max_blocks, max_steps=max_steps)
 
             ev.nphotons = len(ev.photons_beg.pos)
 
@@ -114,7 +111,9 @@ class Simulation(object):
 
         for ev in iterable:
             gpu_photons = gpu.GPUPhotons(ev.photons_beg)
-            gpu.propagate(self.gpu, gpu_photons, self.rng_states, nthreads_per_block=self.nthreads_per_block, max_blocks=self.max_blocks)
+            gpu_photons.propagate(self.gpu_geometry, self.rng_states,
+                                  nthreads_per_block=self.nthreads_per_block,
+                                  max_blocks=self.max_blocks)
             gpu_channels = self.gpu_daq.acquire(gpu_photons, self.rng_states, nthreads_per_block=self.nthreads_per_block, max_blocks=self.max_blocks)
             self.gpu_pdf.add_hits_to_pdf(gpu_channels)
         
@@ -143,7 +142,9 @@ class Simulation(object):
 
         for ev in iterable:
             gpu_photons = gpu.GPUPhotons(ev.photons_beg)
-            gpu.propagate(self.gpu, gpu_photons, self.rng_states, nthreads_per_block=self.nthreads_per_block, max_blocks=self.max_blocks)
+            gpu_photons.propagate(self.gpu_geometry, self.rng_states,
+                                  nthreads_per_block=self.nthreads_per_block,
+                                  max_blocks=self.max_blocks)
             gpu_channels = self.gpu_daq.acquire(gpu_photons, self.rng_states, nthreads_per_block=self.nthreads_per_block, max_blocks=self.max_blocks)
             self.gpu_pdf.accumulate_pdf_eval(gpu_channels)