def init_arrays(self):
super().init_arrays()
# sort arrays
self.sorted_keys = carr.zeros(self.num_particles, dtype=np.int32,
backend=self.backend)
self.indices = carr.zeros(self.num_particles, dtype=np.int32,
backend=self.backend)
def init_arrays(self):
# sort arrays
self.bin_counts = carr.zeros(self.max_key,
dtype=np.int32,
backend=self.backend)
self.start_indices = carr.zeros(self.max_key,
dtype=np.int32,
backend=self.backend)
self.keys = carr.zeros(self.num_particles,
dtype=np.int32,
backend=self.backend)
self.sorted_indices = carr.zeros(self.num_particles,
dtype=np.int32,
backend=self.backend)
# neighbor arrays
self.nbr_lengths = carr.zeros(self.num_particles,
dtype=np.int32,
backend=self.backend)
self.nbr_starts = carr.zeros(self.num_particles,
dtype=np.int32,
backend=self.backend)
self.nbrs = carr.zeros(2 * self.num_particles,
dtype=np.int32,
backend=self.backend)
def test_extend(backend):
check_import(backend)
# Given
dev_array = make_dev_array(backend)
new_array = 2 + array.zeros(64, dtype=np.int32, backend=backend)
# When
dev_array.extend(new_array)
# Then
old_nparr = dev_array.get()
new_nparr = new_array.get()
assert np.all(old_nparr[-len(new_array)] == new_nparr)
def test_add_particles(self, backend):
check_import(backend)
self.setup()
# Given
pa = self.pa
h = DeviceHelper(pa, backend=backend)
# When
pa.set_device_helper(h)
x = array.zeros(4, np.float32, backend=backend)
h.add_particles(x=x)
# Then
assert np.all(np.sort(h.x.get()) == np.array([0., 0., 0., 0., 0., 1.]))
def test_cumsum(backend):
check_import(backend)
# Given
a = array.ones(100, dtype=int, backend=backend)
# When
b = array.cumsum(a)
# Then
a.pull()
b.pull()
assert np.all(b.data == np.cumsum(a.data))
# Test cumsum with an out argument
# Given
out = array.zeros(100, dtype=int, backend=backend)
# When
b = array.cumsum(a, out=out)
# Then
out.pull()
assert np.all(out.data == np.cumsum(a.data))
def init_arrays(self):
super().init_arrays()
self.sort_offsets = carr.zeros(self.num_particles, dtype=np.int32,
backend=self.backend)
