oxirs-vec 0.2.4

Vector index abstractions for semantic similarity and AI-augmented querying
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36

//! Common GPU types and structures

/// GPU device information
#[derive(Debug, Clone)]
pub struct GpuDevice {
    pub device_id: i32,
    pub name: String,
    pub compute_capability: (i32, i32),
    pub total_memory: usize,
    pub free_memory: usize,
    pub max_threads_per_block: i32,
    pub max_blocks_per_grid: i32,
    pub warp_size: i32,
    pub memory_bandwidth: f32,
    pub peak_flops: f64,
}

/// GPU execution configuration
#[derive(Debug, Clone)]
pub struct GpuExecutionConfig {
    pub block_size: u32,
    pub grid_size: u32,
    pub shared_memory_size: u32,
    pub stream_id: Option<u32>,
}

impl Default for GpuExecutionConfig {
    fn default() -> Self {
        Self {
            block_size: 256,
            grid_size: 256,
            shared_memory_size: 0,
            stream_id: None,
        }
    }
}