Skip to content

Conversation

@zhangfengcdt
Copy link
Member

This commit introduces GPU-accelerated spatial join capabilities to SedonaDB, enabling significant performance improvements for large-scale spatial join operations.

Key changes:

  • Add new sedona-spatial-join-gpu crate that provides GPU-accelerated spatial join execution using CUDA via the sedona-libgpuspatial library.
  • Implement GpuSpatialJoinExec execution plan with build/probe phases that efficiently handles partitioned data by sharing build-side data across probes.
  • Add GPU backend abstraction (GpuBackend) for geometry data transfer and spatial predicate evaluation on GPU.
  • Extend the spatial join optimizer to automatically select GPU execution when available and beneficial, with configurable thresholds and fallback to CPU.
  • Add configuration options in SedonaOptions for GPU spatial join settings including enable/disable, row thresholds, and CPU fallback behavior.
  • Include comprehensive benchmarks and functional tests for GPU spatial join correctness validation against CPU reference implementations.

…support

This commit introduces GPU-accelerated spatial join capabilities to SedonaDB,
enabling significant performance improvements for large-scale spatial join
operations.

Key changes:
- Add new `sedona-spatial-join-gpu` crate that provides GPU-accelerated spatial
  join execution using CUDA via the `sedona-libgpuspatial` library.
- Implement `GpuSpatialJoinExec` execution plan with build/probe phases that
  efficiently handles partitioned data by sharing build-side data across probes.
- Add GPU backend abstraction (`GpuBackend`) for geometry data transfer and
  spatial predicate evaluation on GPU.
- Extend the spatial join optimizer to automatically select GPU execution when
  available and beneficial, with configurable thresholds and fallback to CPU.
- Add configuration options in `SedonaOptions` for GPU spatial join settings
  including enable/disable, row thresholds, and CPU fallback behavior.
- Include comprehensive benchmarks and functional tests for GPU spatial join
  correctness validation against CPU reference implementations.
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants