Initial Commit

python/cudf/cudf/_lib/pylibcudf/libcudf/lists/gather.pxd

@@ -10,6 +10,6 @@ from cudf._lib.pylibcudf.libcudf.lists.lists_column_view cimport (
 
 cdef extern from "cudf/lists/gather.hpp" namespace "cudf::lists" nogil:
     cdef unique_ptr[column] segmented_gather(
-        const lists_column_view source_column,
-        const lists_column_view gather_map_list
+        const lists_column_view& source_column,
+        const lists_column_view& gather_map_list
     ) except +

python/cudf/cudf/_lib/pylibcudf/lists.pxd

@@ -2,6 +2,7 @@
 
 from libcpp cimport bool
 
+from cudf._lib.pylibcudf.libcudf.copying cimport out_of_bounds_policy
 from cudf._lib.pylibcudf.libcudf.types cimport size_type
 
 from .column cimport Column
@@ -23,3 +24,5 @@ cpdef Column contains(Column, ColumnOrScalar)
 cpdef Column contains_nulls(Column)
 
 cpdef Column index_of(Column, ColumnOrScalar, bool)
+
+cpdef Column segmented_gather(Column, Column, bool)

python/cudf/cudf/_lib/pylibcudf/lists.pyx

@@ -206,3 +206,9 @@ cpdef Column index_of(Column input, ColumnOrScalar search_key, bool find_first_o
             find_option,
         ))
     return Column.from_libcudf(move(c_result))
+
+
+cpdef Column segmented_gather(Column input, Column gather_map_list, bool check_bounds):
+    """column with elements in list of rows gathered based on gather_map_list
+    """
+    pass