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
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126

#include "datetime_shim.h"
#include <cudf/datetime.hpp>
#include <cudf/utilities/default_stream.hpp>

namespace cudf_shims {

std::unique_ptr<OwnedColumn> extract_year(const OwnedColumn& col) {
    auto result = cudf::datetime::extract_datetime_component(
        col.view(), cudf::datetime::datetime_component::YEAR);
    return std::make_unique<OwnedColumn>(std::move(result));
}

std::unique_ptr<OwnedColumn> extract_month(const OwnedColumn& col) {
    auto result = cudf::datetime::extract_datetime_component(
        col.view(), cudf::datetime::datetime_component::MONTH);
    return std::make_unique<OwnedColumn>(std::move(result));
}

std::unique_ptr<OwnedColumn> extract_day(const OwnedColumn& col) {
    auto result = cudf::datetime::extract_datetime_component(
        col.view(), cudf::datetime::datetime_component::DAY);
    return std::make_unique<OwnedColumn>(std::move(result));
}

std::unique_ptr<OwnedColumn> extract_hour(const OwnedColumn& col) {
    auto result = cudf::datetime::extract_datetime_component(
        col.view(), cudf::datetime::datetime_component::HOUR);
    return std::make_unique<OwnedColumn>(std::move(result));
}

std::unique_ptr<OwnedColumn> extract_minute(const OwnedColumn& col) {
    auto result = cudf::datetime::extract_datetime_component(
        col.view(), cudf::datetime::datetime_component::MINUTE);
    return std::make_unique<OwnedColumn>(std::move(result));
}

std::unique_ptr<OwnedColumn> extract_second(const OwnedColumn& col) {
    auto result = cudf::datetime::extract_datetime_component(
        col.view(), cudf::datetime::datetime_component::SECOND);
    return std::make_unique<OwnedColumn>(std::move(result));
}

std::unique_ptr<OwnedColumn> extract_weekday(const OwnedColumn& col) {
    auto result = cudf::datetime::extract_datetime_component(
        col.view(), cudf::datetime::datetime_component::WEEKDAY);
    return std::make_unique<OwnedColumn>(std::move(result));
}

std::unique_ptr<OwnedColumn> extract_day_of_year(const OwnedColumn& col) {
    auto result = cudf::datetime::day_of_year(col.view());
    return std::make_unique<OwnedColumn>(std::move(result));
}

std::unique_ptr<OwnedColumn> last_day_of_month(const OwnedColumn& col) {
    auto stream = cudf::get_default_stream();
    auto mr = cudf::get_current_device_resource_ref();
    auto result = cudf::datetime::last_day_of_month(col.view(), stream, mr);
    return std::make_unique<OwnedColumn>(std::move(result));
}

std::unique_ptr<OwnedColumn> add_calendrical_months_scalar(
    const OwnedColumn& col, const OwnedScalar& months)
{
    auto stream = cudf::get_default_stream();
    auto mr = cudf::get_current_device_resource_ref();
    auto result = cudf::datetime::add_calendrical_months(
        col.view(), *months.inner, stream, mr);
    return std::make_unique<OwnedColumn>(std::move(result));
}

std::unique_ptr<OwnedColumn> add_calendrical_months_column(
    const OwnedColumn& col, const OwnedColumn& months)
{
    auto stream = cudf::get_default_stream();
    auto mr = cudf::get_current_device_resource_ref();
    auto result = cudf::datetime::add_calendrical_months(
        col.view(), months.view(), stream, mr);
    return std::make_unique<OwnedColumn>(std::move(result));
}

std::unique_ptr<OwnedColumn> is_leap_year(const OwnedColumn& col) {
    auto stream = cudf::get_default_stream();
    auto mr = cudf::get_current_device_resource_ref();
    auto result = cudf::datetime::is_leap_year(col.view(), stream, mr);
    return std::make_unique<OwnedColumn>(std::move(result));
}

std::unique_ptr<OwnedColumn> days_in_month(const OwnedColumn& col) {
    auto stream = cudf::get_default_stream();
    auto mr = cudf::get_current_device_resource_ref();
    auto result = cudf::datetime::days_in_month(col.view(), stream, mr);
    return std::make_unique<OwnedColumn>(std::move(result));
}

std::unique_ptr<OwnedColumn> extract_quarter(const OwnedColumn& col) {
    auto stream = cudf::get_default_stream();
    auto mr = cudf::get_current_device_resource_ref();
    auto result = cudf::datetime::extract_quarter(col.view(), stream, mr);
    return std::make_unique<OwnedColumn>(std::move(result));
}

std::unique_ptr<OwnedColumn> ceil_datetimes(const OwnedColumn& col, int32_t freq) {
    auto stream = cudf::get_default_stream();
    auto mr = cudf::get_current_device_resource_ref();
    auto f = static_cast<cudf::datetime::rounding_frequency>(freq);
    auto result = cudf::datetime::ceil_datetimes(col.view(), f, stream, mr);
    return std::make_unique<OwnedColumn>(std::move(result));
}

std::unique_ptr<OwnedColumn> floor_datetimes(const OwnedColumn& col, int32_t freq) {
    auto stream = cudf::get_default_stream();
    auto mr = cudf::get_current_device_resource_ref();
    auto f = static_cast<cudf::datetime::rounding_frequency>(freq);
    auto result = cudf::datetime::floor_datetimes(col.view(), f, stream, mr);
    return std::make_unique<OwnedColumn>(std::move(result));
}

std::unique_ptr<OwnedColumn> round_datetimes(const OwnedColumn& col, int32_t freq) {
    auto stream = cudf::get_default_stream();
    auto mr = cudf::get_current_device_resource_ref();
    auto f = static_cast<cudf::datetime::rounding_frequency>(freq);
    auto result = cudf::datetime::round_datetimes(col.view(), f, stream, mr);
    return std::make_unique<OwnedColumn>(std::move(result));
}

} // namespace cudf_shims