GPU aspects

Author: xbuffat

xfields/beam_elements/waketracker/convolution.py

@@ -118,8 +118,8 @@ def _initialize_conv_data(self, _flatten=False, moments_data=None, beta0=None):
 
             # only positive frequencies because we are using rfft
             phase_term = self._context.nparray_to_context_array(np.exp(
-                1j * 2 * np.pi * np.arange(self._M_aux//2 + 1) *
-                ((self._N_S - 1) * self._N_aux + self._N_1) / self._M_aux))
+                1j * 2 * np.pi * np.arange(int(self._M_aux)//2 + 1) *
+                ((int(self._N_S) - 1) * int(self._N_aux) + int(self._N_1)) / int(self._M_aux)))
 
         else:
             raise NotImplementedError('Flattened wakes are not implemented yet')
@@ -253,7 +253,7 @@ def _build_z_wake(z_a, z_b, num_turns, n_aux, m_aux, circumference, dz,
 
     z_c = z_a  # For wakefield, z_c = z_a
     z_d = z_b  # For wakefield, z_d = z_b
-    z_wake = np.zeros((num_turns, m_aux))
+    z_wake = np.zeros((int(num_turns), int(m_aux)))
     for tt in range(num_turns):
         z_a_turn = z_a + tt * circumference
         z_b_turn = z_b + tt * circumference

xfields/slicers/compressed_profile.py

@@ -119,8 +119,8 @@ def __init__(self,
 
         self.moments_names = moments
 
-        self.xoinitialize(_N_T=_N_T,_first_target_slot=first_target_slot, _N_aux=_N_aux, num_turns=num_turns, 
-                          data=(len(moments), num_turns, self._M_aux),
+        self.xoinitialize(_N_T=int(_N_T),_first_target_slot=int(first_target_slot), _N_aux=int(_N_aux), num_turns=int(num_turns), 
+                          data=(len(moments), int(num_turns), int(self._M_aux)),
                           **kwargs)
 
 

xfields/slicers/uniform.py

@@ -348,7 +348,7 @@ def _reshape_for_multibunch(self, data):
     def _to_npbuffer(self):
         tmp_buffer = self._buffer.buffer[self._offset:
                                    self._offset + self._xobject._size]
-        return self._context.nparray_from_context_array(tmp_buffer)
+        return self._context.nparray_from_context_array(tmp_buffer).astype(np.int8)
 
     @classmethod
     def _from_npbuffer(cls, buffer):