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

#pragma once

#include <cudf/aggregation.hpp>
#include <cudf/types.hpp>
#include <memory>
#include "rust/cxx.h"

namespace cudf_shims {

/// Owning wrapper around a cudf::groupby_aggregation.
struct OwnedAggregation {
    std::unique_ptr<cudf::groupby_aggregation> inner;

    explicit OwnedAggregation(std::unique_ptr<cudf::groupby_aggregation> agg)
        : inner(std::move(agg)) {}
};

// ── Factory functions ──────────────────────────────────────────

std::unique_ptr<OwnedAggregation> agg_sum();
std::unique_ptr<OwnedAggregation> agg_product();
std::unique_ptr<OwnedAggregation> agg_min();
std::unique_ptr<OwnedAggregation> agg_max();
std::unique_ptr<OwnedAggregation> agg_count(int32_t null_handling);
std::unique_ptr<OwnedAggregation> agg_any();
std::unique_ptr<OwnedAggregation> agg_all();
std::unique_ptr<OwnedAggregation> agg_sum_of_squares();
std::unique_ptr<OwnedAggregation> agg_mean();
std::unique_ptr<OwnedAggregation> agg_median();
std::unique_ptr<OwnedAggregation> agg_variance(int32_t ddof);
std::unique_ptr<OwnedAggregation> agg_std(int32_t ddof);
std::unique_ptr<OwnedAggregation> agg_nunique(int32_t null_handling);
std::unique_ptr<OwnedAggregation> agg_nth_element(int32_t n, int32_t null_handling);
std::unique_ptr<OwnedAggregation> agg_collect_list(int32_t null_handling);
std::unique_ptr<OwnedAggregation> agg_collect_set(int32_t null_handling);
std::unique_ptr<OwnedAggregation> agg_argmax();
std::unique_ptr<OwnedAggregation> agg_argmin();
std::unique_ptr<OwnedAggregation> agg_row_number();
std::unique_ptr<OwnedAggregation> agg_quantile(double q);
std::unique_ptr<OwnedAggregation> agg_lag(int32_t offset);
std::unique_ptr<OwnedAggregation> agg_lead(int32_t offset);

} // namespace cudf_shims