waxmorph.simulator.count_neighbors_step

waxmorph.simulator.count_neighbors_step(X, R, CT, particle_count, n_tot, n_epi, n_mes, device='cuda', grid=None)

Atomically accumulate into caller-zeroed neighbor counters.

Parameters:

• X (array)

• R (array)

• CT (array)

• particle_count (int)

• n_tot (array)

• n_epi (array)

• n_mes (array)

• device (str)

• grid (HashGrid | None)

Return type:

None