Merge pull request #96 from PyCOMPLETE/hotfix/aperture-slicemonitor-fix

aperture: fixing bug with aperture on GPU...

Author: (Py)thon (Co)llective (M)acro-(P)article Simulation (L)ibrary with (E)xtendible (T)racking (E)lements

PyHEADTAIL/aperture/aperture.py

@@ -19,7 +19,6 @@
 import numpy as np
 
 def make_int32(array):
-    # return np.array(array, dtype=np.int32)
     return array.astype(np.int32)
 
 
@@ -89,8 +88,9 @@ def relocate_lost_particles(beam, alive):
 
         beam.reorder(perm)
 
-        n_alive = make_int32(pm.sum(alive))
-        return n_alive
+        # on CPU: (even if pm.device == 'GPU', as pm.sum returns np.ndarray)
+        n_alive = pm.sum(alive)
+        return np.int32(n_alive)
 
 
 class RectangularApertureX(Aperture):