vibespatial.spatial.point_distance

Functions

compute_distance_center_device(query_owned, tree_owned)	Reduce device row bounds to one device-resident centering vector.
compute_point_distance_gpu(→ bool)	Compute point -> geometry distance on device for a single tree family.
compute_pointset_distance_gpu(→ bool)	Run one point/multipoint family span from a shared relation partition.
supported_point_distance_families(...)	Return the set of tree families supported by GPU point-distance kernels.
supported_pointset_distance_families(...)	Return all owned target families supported by pointset distance.

Module Contents

vibespatial.spatial.point_distance.compute_distance_center_device(query_owned: vibespatial.geometry.owned.OwnedGeometryArray, tree_owned: vibespatial.geometry.owned.OwnedGeometryArray)

Reduce device row bounds to one device-resident centering vector.

vibespatial.spatial.point_distance.compute_point_distance_gpu(query_owned: vibespatial.geometry.owned.OwnedGeometryArray, tree_owned: vibespatial.geometry.owned.OwnedGeometryArray, d_left, d_right, d_distances, pair_count: int, *, tree_family: vibespatial.geometry.buffers.GeometryFamily, exclusive: bool = False, compute_precision: vibespatial.runtime.precision.PrecisionMode = PrecisionMode.AUTO, logical_count=None, center_device=None) → bool

Compute point -> geometry distance on device for a single tree family.

Writes results into d_distances (device float64 array, shape pair_count). When logical_count is supplied, pair_count is physical capacity and the one-element device count guards the active compact prefix in-kernel. Returns True if the kernel was dispatched, False if the family is not supported (caller should fall back to Shapely).

vibespatial.spatial.point_distance.compute_pointset_distance_gpu(query_owned: vibespatial.geometry.owned.OwnedGeometryArray, tree_owned: vibespatial.geometry.owned.OwnedGeometryArray, d_left, d_right, d_distances, launch_capacity: int, *, query_family: vibespatial.geometry.buffers.GeometryFamily, tree_family: vibespatial.geometry.buffers.GeometryFamily, source_offset, logical_count, source_positions, center_device, exclusive: bool = False, compute_precision: vibespatial.runtime.precision.PrecisionMode = PrecisionMode.AUTO) → bool

Run one point/multipoint family span from a shared relation partition.

vibespatial.spatial.point_distance.supported_point_distance_families() → frozenset[vibespatial.geometry.buffers.GeometryFamily]

Return the set of tree families supported by GPU point-distance kernels.

vibespatial.spatial.point_distance.supported_pointset_distance_families() → frozenset[vibespatial.geometry.buffers.GeometryFamily]

Return all owned target families supported by pointset distance.