Module functions 

Auto-generated module

🤖 Generated with SplitRS

Functions

compute_bounding_box

Compute the bounding box of a set of positions.

distribute_cells_to_gpus

Distribute n_cells cells evenly across n_gpus GPUs.

gpu_neighbor_search_kernel

Return all particle pairs (i, j) with i < j and dist(i, j) < cutoff.

gpu_prefix_sum

Exclusive prefix sum (scan) of counts.

insert_particles

Insert particles into an existing cell list without a full rebuild.

morton_decode

Decode a 30-bit Morton code back to 3D integer coordinates.

morton_encode

Compute a 30-bit Morton code from 3D integer coordinates.

morton_sort

Compute Morton codes for a set of positions and sort indices by Morton code.

parallel_count_particles

Count how many particles fall into each cell of a regular grid.

parallel_morton_sort

Sort particle indices by Morton code using Rayon for parallel code generation.

parallel_prefix_sum

Compute an exclusive prefix sum (scan) over counts.

position_to_morton

Compute the 30-bit Morton code for a position relative to a bounding box.

query_neighbors

Query all particles within radius of query_pos from a set of positions, using a pre-built GpuCellList.

radix_sort_mock

Mock radix sort: stable sort of keys returning (sorted_keys, sorted_indices).

reorder_by_permutation

Reorder an array according to a permutation.