cudf-cxx 0.2.0

cxx-based FFI bridge between Rust and NVIDIA libcudf C++ API
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
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71

#pragma once

#include <cudf/reduction.hpp>
#include <cudf/types.hpp>
#include <memory>
#include "rust/cxx.h"
#include "column_shim.h"
#include "scalar_shim.h"

namespace cudf_shims {

/// Reduce a column to a single scalar value.
std::unique_ptr<OwnedScalar> reduce(
    const OwnedColumn& col,
    int32_t agg_kind,
    int32_t output_type_id);

/// Prefix scan (cumulative operation).
std::unique_ptr<OwnedColumn> scan(
    const OwnedColumn& col,
    int32_t agg_kind,
    bool inclusive);

/// Segmented reduce within segments defined by offsets.
std::unique_ptr<OwnedColumn> segmented_reduce(
    const OwnedColumn& col,
    const OwnedColumn& offsets,
    int32_t agg_kind,
    int32_t output_type_id,
    bool include_nulls);

/// Compute the minimum and maximum of a column simultaneously.
/// Returns a 2-element OwnedTable where column 0 is the min scalar as a column,
/// and column 1 is the max scalar as a column.
/// Actually returns two scalars packed into a pair struct.
struct MinMaxResult {
    std::unique_ptr<OwnedScalar> min_val;
    std::unique_ptr<OwnedScalar> max_val;
};

std::unique_ptr<MinMaxResult> minmax(const OwnedColumn& col);

/// Move min out of MinMaxResult.
std::unique_ptr<OwnedScalar> minmax_take_min(MinMaxResult& result);

/// Move max out of MinMaxResult.
std::unique_ptr<OwnedScalar> minmax_take_max(MinMaxResult& result);

/// Reduce with an initial value.
std::unique_ptr<OwnedScalar> reduce_with_init(
    const OwnedColumn& col,
    int32_t agg_kind,
    int32_t output_type_id,
    const OwnedScalar& init);

/// Check if a reduction aggregation is valid for a given source data type.
bool is_valid_reduction_aggregation(int32_t source_type_id, int32_t agg_kind);

/// Reduce: variance with custom ddof.
std::unique_ptr<OwnedScalar> reduce_var_with_ddof(
    const OwnedColumn& col,
    int32_t ddof,
    int32_t output_type_id);

/// Reduce: standard deviation with custom ddof.
std::unique_ptr<OwnedScalar> reduce_std_with_ddof(
    const OwnedColumn& col,
    int32_t ddof,
    int32_t output_type_id);

} // namespace cudf_shims