vibespatial.spatial.point_distance

Functions

compute_point_distance_gpu(→ bool)	Compute point -> geometry distance on device for a single tree family.
supported_point_distance_families(...)	Return the set of tree families supported by GPU point-distance kernels.

Module Contents

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) → bool

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

Writes results into d_distances (device float64 array, shape pair_count). Returns True if the kernel was dispatched, False if the family is not supported (caller should fall back to Shapely).

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.