Skip to content

Commit

Permalink
Support nested types in lists::contains (#10548)
Browse files Browse the repository at this point in the history
This extends the `lists::contains` API to support nested types (lists + structs) with arbitrarily nested levels. As such, `lists::contains` will work with literally any type of input data.

In addition, the related implementation has been significantly refactored to facilitate adding new implementation.

Closes #8958.
Depends on:
 * #10730
 * #10883
 * #10999
 * #11019
 * #11037

Authors:
  - Nghia Truong (https://github.com/ttnghia)
  - Bradley Dice (https://github.com/bdice)

Approvers:
  - MithunR (https://github.com/mythrocks)
  - Bradley Dice (https://github.com/bdice)

URL: #10548
  • Loading branch information
ttnghia authored Aug 22, 2022
1 parent a5dcb32 commit 8f3cc74
Show file tree
Hide file tree
Showing 3 changed files with 1,144 additions and 220 deletions.
7 changes: 7 additions & 0 deletions cpp/include/cudf/lists/list_device_view.cuh
Original file line number Diff line number Diff line change
Expand Up @@ -129,6 +129,13 @@ class list_device_view {
*/
[[nodiscard]] __device__ inline size_type size() const { return _size; }

/**
* @brief Returns the row index of this list in the original lists column.
*
* @return The row index of this list
*/
[[nodiscard]] __device__ inline size_type row_index() const { return _row_index; }

/**
* @brief Fetches the lists_column_device_view that contains this list.
*
Expand Down
Loading

0 comments on commit 8f3cc74

Please sign in to comment.