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

#include "strings/replace_shim.h"
#include <cudf/strings/replace.hpp>
#include <cudf/strings/replace_re.hpp>
#include <cudf/scalar/scalar.hpp>
#include <cudf/scalar/scalar_factories.hpp>
#include <cudf/strings/regex/regex_program.hpp>
#include <cudf/utilities/default_stream.hpp>
#include <string>

namespace cudf_shims {

std::unique_ptr<OwnedColumn> str_replace(
    const OwnedColumn& col, rust::Str target, rust::Str replacement)
{
    auto stream = cudf::get_default_stream();
    auto mr = cudf::get_current_device_resource_ref();
    std::string tgt(target.data(), target.size());
    std::string repl(replacement.data(), replacement.size());
    cudf::string_scalar scalar_target(tgt, true, stream);
    cudf::string_scalar scalar_repl(repl, true, stream);
    auto result = cudf::strings::replace(
        col.view(), scalar_target, scalar_repl, -1, stream, mr);
    return std::make_unique<OwnedColumn>(std::move(result));
}

std::unique_ptr<OwnedColumn> str_replace_re(
    const OwnedColumn& col, rust::Str pattern, rust::Str replacement)
{
    auto stream = cudf::get_default_stream();
    auto mr = cudf::get_current_device_resource_ref();
    std::string pat(pattern.data(), pattern.size());
    std::string repl(replacement.data(), replacement.size());
    auto prog = cudf::strings::regex_program::create(pat);
    cudf::string_scalar scalar_repl(repl, true, stream);
    auto result = cudf::strings::replace_re(
        col.view(), *prog, scalar_repl, std::nullopt, stream, mr);
    return std::make_unique<OwnedColumn>(std::move(result));
}

std::unique_ptr<OwnedColumn> str_replace_slice(
    const OwnedColumn& col, rust::Str replacement, int32_t start, int32_t stop)
{
    auto stream = cudf::get_default_stream();
    auto mr = cudf::get_current_device_resource_ref();
    std::string repl(replacement.data(), replacement.size());
    cudf::string_scalar scalar_repl(repl, true, stream);
    auto result = cudf::strings::replace_slice(
        col.view(), scalar_repl, start, stop, stream, mr);
    return std::make_unique<OwnedColumn>(std::move(result));
}

std::unique_ptr<OwnedColumn> str_replace_multiple(
    const OwnedColumn& col, const OwnedColumn& targets, const OwnedColumn& replacements)
{
    auto stream = cudf::get_default_stream();
    auto mr = cudf::get_current_device_resource_ref();
    auto result = cudf::strings::replace_multiple(
        col.view(),
        cudf::strings_column_view(targets.view()),
        cudf::strings_column_view(replacements.view()),
        stream, mr);
    return std::make_unique<OwnedColumn>(std::move(result));
}

std::unique_ptr<OwnedColumn> str_replace_with_backrefs(
    const OwnedColumn& col, rust::Str pattern, rust::Str replacement)
{
    auto stream = cudf::get_default_stream();
    auto mr = cudf::get_current_device_resource_ref();
    std::string pat(pattern.data(), pattern.size());
    std::string repl(replacement.data(), replacement.size());
    auto prog = cudf::strings::regex_program::create(pat);
    auto result = cudf::strings::replace_with_backrefs(
        col.view(), *prog, repl, stream, mr);
    return std::make_unique<OwnedColumn>(std::move(result));
}

std::unique_ptr<OwnedColumn> str_replace_re_multiple(
    const OwnedColumn& col,
    rust::Slice<const rust::String> patterns,
    const OwnedColumn& replacements)
{
    auto stream = cudf::get_default_stream();
    auto mr = cudf::get_current_device_resource_ref();
    std::vector<std::string> pat_vec;
    pat_vec.reserve(patterns.size());
    for (const auto& p : patterns) {
        pat_vec.emplace_back(p.data(), p.size());
    }
    auto result = cudf::strings::replace_re(
        col.view(), pat_vec,
        cudf::strings_column_view(replacements.view()),
        cudf::strings::regex_flags::DEFAULT,
        stream, mr);
    return std::make_unique<OwnedColumn>(std::move(result));
}

} // namespace cudf_shims