algeon-datafusion 0.1.2

DataFusion physical-plan adapter for native GPU execution.
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

use super::CuvsOperation;
use crate::{error::Error, gpu_functions::dense_relation_errors::DenseRelationErrorContext};

pub(super) const ERRORS: DenseRelationErrorContext = DenseRelationErrorContext {
    family: "cuVS",
    operation_fact: "cuvs_operation",
    validate_operation: "algeon_datafusion.cuvs_sql.validate",
    validate_schema_operation: "algeon_datafusion.cuvs_sql.validate_schema",
    execute_operation: "algeon_datafusion.cuvs_sql.execute",
    required_feature: "cuvs",
    feature_disabled_message: "cuVS vector algorithm execution requires the cuvs Cargo feature",
};

pub(super) fn invalid_call(operation: CuvsOperation, reason: &'static str, message: impl Into<String>) -> Error {
    ERRORS.invalid_call(operation.descriptor().sql_function_name, reason, message)
}

pub(super) fn invalid_option(
    operation: CuvsOperation,
    reason: &'static str,
    argument: Option<&'static str>,
    message: impl Into<String>,
) -> Error {
    ERRORS.invalid_option(operation.descriptor().sql_function_name, reason, argument, message)
}

pub(super) fn field_not_found(operation: CuvsOperation, relation_role: &'static str, field: &str) -> Error {
    ERRORS.field_not_found(operation.descriptor().sql_function_name, relation_role, field)
}

pub(super) fn unsupported_type(
    operation: CuvsOperation,
    relation_role: &'static str,
    field: &str,
    actual: impl ToString,
    expected: impl ToString,
) -> Error {
    ERRORS.unsupported_type(operation.descriptor().sql_function_name, relation_role, field, actual, expected)
}

pub(super) fn feature_disabled(operation: CuvsOperation) -> Error {
    ERRORS.feature_disabled(operation.descriptor().sql_function_name)
}