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

#include "structs_shim.h"
#include <cudf/structs/structs_column_view.hpp>
#include <cudf/copying.hpp>
#include <cudf/utilities/default_stream.hpp>
#include <stdexcept>

namespace cudf_shims {

std::unique_ptr<OwnedColumn> structs_extract(
    const OwnedColumn& col,
    int32_t index)
{
    auto structs_view = cudf::structs_column_view(col.view());

    if (index < 0 || index >= structs_view.num_children()) {
        throw std::out_of_range(
            "structs_extract: index " + std::to_string(index) +
            " out of range [0, " + std::to_string(structs_view.num_children()) + ")");
    }

    auto child_view = structs_view.get_sliced_child(index);
    // Materialize a copy so the returned column owns its own memory.
    auto result = std::make_unique<cudf::column>(child_view);
    return std::make_unique<OwnedColumn>(std::move(result));
}

} // namespace cudf_shims