cudf-cxx 0.3.1

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

#include "search_shim.h"
#include "order_utils.h"
#include "scalar_shim.h"
#include <cudf/search.hpp>
#include <cudf/utilities/default_stream.hpp>
#include <vector>

namespace cudf_shims {

std::unique_ptr<OwnedColumn> lower_bound(
    const OwnedTable& table,
    const OwnedTable& values,
    rust::Slice<const int32_t> orders,
    rust::Slice<const int32_t> null_orders)
{
    auto col_order = to_column_order(orders);
    auto nul_order = to_null_order(null_orders);

    auto result = cudf::lower_bound(
        table.view(),
        values.view(),
        col_order,
        nul_order);

    return std::make_unique<OwnedColumn>(std::move(result));
}

std::unique_ptr<OwnedColumn> upper_bound(
    const OwnedTable& table,
    const OwnedTable& values,
    rust::Slice<const int32_t> orders,
    rust::Slice<const int32_t> null_orders)
{
    auto col_order = to_column_order(orders);
    auto nul_order = to_null_order(null_orders);

    auto result = cudf::upper_bound(
        table.view(),
        values.view(),
        col_order,
        nul_order);

    return std::make_unique<OwnedColumn>(std::move(result));
}

std::unique_ptr<OwnedColumn> contains_column(
    const OwnedColumn& haystack,
    const OwnedColumn& needles)
{
    auto result = cudf::contains(haystack.view(), needles.view());
    return std::make_unique<OwnedColumn>(std::move(result));
}

bool contains_scalar(
    const OwnedColumn& haystack,
    const OwnedScalar& needle)
{
    return cudf::contains(haystack.view(), *needle.inner);
}

} // namespace cudf_shims