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

#pragma once

#include <cudf/quantiles.hpp>
#include <cudf/types.hpp>
#include <memory>
#include <vector>
#include "rust/cxx.h"
#include "column_shim.h"
#include "table_shim.h"

namespace cudf_shims {

/// Compute quantile(s) of a column.
std::unique_ptr<OwnedColumn> quantile(
    const OwnedColumn& col,
    rust::Slice<const double> q,
    int32_t interp);

/// Compute quantiles of a table (row-wise).
std::unique_ptr<OwnedTable> quantiles_table(
    const OwnedTable& table,
    rust::Slice<const double> q,
    int32_t interp,
    bool is_input_sorted,
    rust::Slice<const int32_t> orders,
    rust::Slice<const int32_t> null_orders);

/// Compute percentile approximation using t-digest.
std::unique_ptr<OwnedColumn> percentile_approx(
    const OwnedColumn& tdigest_col,
    rust::Slice<const double> percentiles);

} // namespace cudf_shims