algeon-cudf 0.1.2

Safe libcudf execution component for Algeon
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
37
38
39
40
41
42
43
44
45
46
47
48

use crate::{
    core::{column::Column, error::Result, types::CudfType},
    runtime::Runtime,
};

mod private {
    use crate::core::types::CudfType;

    pub trait Sealed {}

    impl<T: CudfType> Sealed for T {}
    impl Sealed for &str {}
}

/// Host slice element types supported by the high-level `Series` and `Runtime`
/// constructors.
///
/// This trait is sealed because host-to-device construction semantics are part
/// of cudf's safety and performance contract.
pub trait HostColumnData: private::Sealed {
    /// Create a GPU-backed column inside an explicit runtime.
    fn create_column_in_runtime(runtime: &Runtime, data: &[Self]) -> Result<Column>
    where Self: Sized;

    /// Create a nullable GPU-backed column inside an explicit runtime.
    fn create_column_nullable_in_runtime(runtime: &Runtime, data: &[Self], validity: &[bool]) -> Result<Column>
    where Self: Sized;
}

impl<T: CudfType> HostColumnData for T {
    fn create_column_in_runtime(runtime: &Runtime, data: &[Self]) -> Result<Column> {
        Column::from_slice_with_runtime(runtime, data)
    }

    fn create_column_nullable_in_runtime(runtime: &Runtime, data: &[Self], validity: &[bool]) -> Result<Column> {
        Column::from_slice_nullable_with_runtime(runtime, data, validity)
    }
}

impl HostColumnData for &str {
    fn create_column_in_runtime(runtime: &Runtime, data: &[Self]) -> Result<Column> {
        Column::from_strs_with_runtime(runtime, data)
    }

    fn create_column_nullable_in_runtime(runtime: &Runtime, data: &[Self], validity: &[bool]) -> Result<Column> {
        Column::from_strs_nullable_with_runtime(runtime, data, validity)
    }
}