libcudf  24.04.00
lists/filling.hpp
Go to the documentation of this file.
1 /*
2  * Copyright (c) 2021-2023, NVIDIA CORPORATION.
3  *
4  * Licensed under the Apache License, Version 2.0 (the "License");
5  * you may not use this file except in compliance with the License.
6  * You may obtain a copy of the License at
7  *
8  * http://www.apache.org/licenses/LICENSE-2.0
9  *
10  * Unless required by applicable law or agreed to in writing, software
11  * distributed under the License is distributed on an "AS IS" BASIS,
12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13  * See the License for the specific language governing permissions and
14  * limitations under the License.
15  */
16 
17 #pragma once
18 
19 #include <cudf/types.hpp>
21 
22 #include <rmm/cuda_stream_view.hpp>
23 #include <rmm/mr/device/per_device_resource.hpp>
24 
25 #include <memory>
26 
27 namespace cudf::lists {
66 std::unique_ptr<column> sequences(
67  column_view const& starts,
68  column_view const& sizes,
69  rmm::cuda_stream_view stream = cudf::get_default_stream(),
70  rmm::mr::device_memory_resource* mr = rmm::mr::get_current_device_resource());
71 
107 std::unique_ptr<column> sequences(
108  column_view const& starts,
109  column_view const& steps,
110  column_view const& sizes,
111  rmm::cuda_stream_view stream = cudf::get_default_stream(),
112  rmm::mr::device_memory_resource* mr = rmm::mr::get_current_device_resource());
113  // end of group
115 } // namespace cudf::lists
A non-owning, immutable view of device data as a column of elements, some of which may be null as ind...
rmm::cuda_stream_view const get_default_stream()
Get the current default stream.
std::unique_ptr< column > sequences(column_view const &starts, column_view const &sizes, rmm::cuda_stream_view stream=cudf::get_default_stream(), rmm::mr::device_memory_resource *mr=rmm::mr::get_current_device_resource())
Create a lists column in which each row contains a sequence of values specified by a tuple of (start,...
Lists column APIs.
Type declarations for libcudf.