`nvalchemi.distributed`.ParticleHaloConfig#

class nvalchemi.distributed.ParticleHaloConfig(ghost_width, partitioner, mesh)[source]#

Configuration for particle-based halo exchange.

Initialized once (from DomainConfig + SpatialPartitioner) and reused across steps.

Parameters:

ghost_width (float) – Halo region width (typically cutoff + skin).
partitioner (SpatialPartitioner) – Spatial grid partitioner (provides cell, pbc, rank bounds). Any-typed here because SpatialPartitioner is defined in nvalchemi.distributed.partitioner.
mesh (DeviceMesh) – 1D device mesh for communication.

property neighbor_ranks: list[int]#: Ranks this one exchanges ghosts with, for the current geometry.

property pbc_shifts: dict[tuple[int, int], list[Float[Tensor, '3']]]#: Materialize current Cartesian shifts from the cached lattice images.