Struct GpuManager 

pub struct GpuManager { /* private fields */ }

Manager for GPU-accelerated operations.

Implementations

impl GpuManager

Aggregation operations impl block

pub fn sum(&self, table: &str, column: &str) -> DbxResult<i64>

SUM aggregation on GPU with configurable reduction strategy.

pub fn count(&self, table: &str, column: &str) -> DbxResult<u64>

COUNT aggregation on GPU (single-pass for simplicity).

pub fn min_max( &self, table: &str, column: &str, find_max: bool, ) -> DbxResult<i32>

MIN/MAX aggregation on GPU.

pub fn filter_gt( &self, table: &str, column: &str, threshold: i32, ) -> DbxResult<Vec<u8>>

Filter GT on GPU. Returns a bitmask (Vec<u8> where 1 means true).

impl GpuManager

GROUP BY operations impl block

pub fn group_by_sum( &self, table: &str, group_column: &str, value_column: &str, ) -> DbxResult<Vec<(i32, i64, i32)>>

GROUP BY with SUM aggregation on GPU. Returns Vec<(group_key, sum_value, count)>

pub fn group_by_count( &self, table: &str, group_column: &str, ) -> DbxResult<Vec<(i32, i32)>>

GROUP BY with COUNT aggregation on GPU. Returns Vec<(group_key, count)>

pub fn group_by_min_max( &self, table: &str, group_column: &str, value_column: &str, find_max: bool, ) -> DbxResult<Vec<(i32, i32, i32)>>

GROUP BY with MIN/MAX aggregation on GPU. Returns Vec<(group_key, min_or_max_value, count)>

impl GpuManager

Hash Join operations impl block

pub fn hash_join( &self, build_table: &str, build_key_column: &str, probe_table: &str, probe_key_column: &str, ) -> DbxResult<Vec<(i32, i32)>>

Hash Join on GPU: Build + Probe phases. Returns Vec<(probe_row_id, build_row_id)> for matched rows.

impl GpuManager

pub fn try_new() -> Option<Self>

Create a new GpuManager. Returns None if GPU acceleration is disabled or if no compatible device is found.

pub fn set_hash_strategy(&mut self, strategy: GpuHashStrategy)

Set GPU hash strategy for GROUP BY operations

pub fn hash_strategy(&self) -> GpuHashStrategy

Get current GPU hash strategy

pub fn set_reduction_strategy(&mut self, strategy: GpuReductionStrategy)

Set GPU reduction strategy for SUM operations

pub fn reduction_strategy(&self) -> GpuReductionStrategy

Get current GPU reduction strategy

pub fn upload_batch(&self, table: &str, batch: &RecordBatch) -> DbxResult<()>

Upload a RecordBatch to GPU memory and cache it.

pub fn upload_batch_pinned( &self, table: &str, batch: &RecordBatch, ) -> DbxResult<()>

Upload a RecordBatch to GPU memory using Pinned Memory for faster DMA transfer.

pub fn clear_table_cache(&self, table: &str)

pub fn clear_all_cache(&self)

impl GpuManager

This impl block contains no items.

Radix Sort operations impl block