Add back get_element

Author: mroeschke

python/morpheus/morpheus/_lib/cudf_helpers.pyx

@@ -35,10 +35,15 @@ from cudf._lib.column cimport Column
 
 # isort: off
 
-# imports needed for from_column_view_with_fix
-import pylibcudf as plc
-from pylibcudf cimport Column as plc_Column, Table as plc_Table
+# imports needed for get_element, which is required by from_column_view_with_fix
+cimport pylibcudf.libcudf.copying as cpp_copying
 from pylibcudf.libcudf.column.column_view cimport column_view
+from libcpp.memory cimport make_unique, unique_ptr
+from pylibcudf.libcudf.scalar.scalar cimport scalar
+from pylibcudf cimport Table as plc_Table, Scalar as plc_Scalar
+import pylibcudf as plc
+
+# imports needed for from_column_view_with_fix
 import rmm
 from libc.stdint cimport uintptr_t
 from cudf.core.buffer import (
@@ -59,6 +64,18 @@ from cudf._lib.null_mask import bitmask_allocation_size_bytes
 
 # isort: on
 
+cdef get_element(column_view col_view, size_type index):
+
+    cdef unique_ptr[scalar] c_output
+    with nogil:
+        c_output = move(
+            cpp_copying.get_element(col_view, index)
+        )
+
+    plc_scalar = plc_Scalar.from_libcudf(move(c_output))
+    return plc.interop.to_arrow(plc_scalar).to_py()
+
+
 cdef Column from_column_view_with_fix(column_view cv, object owner):
     """
     Given a ``cudf::column_view``, constructs a ``cudf.Column`` from it,
@@ -101,13 +118,9 @@ cdef Column from_column_view_with_fix(column_view cv, object owner):
             if offset_child_column.size() == 0:
                 base_nbytes = 0
             else:
-                offset_child_column_owner = owner.children[offsets_column_index]
-                plc_owner = offset_child_column_owner.to_pylibcudf(mode="read")
-                plc_offsets_col = plc_Column.from_column_view(
-                    offset_child_column, plc_owner
-                )
-                plc_scalar = plc.copying.get_element(plc_offsets_col, offset_child_column.size()-1)
-                base_nbytes = plc.interop.to_arrow(plc_scalar).as_py()
+                chars_size = get_element(
+                    offset_child_column, offset_child_column.size()-1)
+                base_nbytes = chars_size
 
     if data_ptr:
         if data_owner is None: