clean up

rapidsai · Nov 14, 2024 · 4acfbde · 4acfbde
1 parent 99ba0f1
commit 4acfbde
Showing 1 changed file with 2 additions and 5 deletions.
diff --git a/python/cudf/cudf/_lib/copying.pyx b/python/cudf/cudf/_lib/copying.pyx
@@ -26,8 +26,6 @@ from pylibcudf.libcudf.types cimport size_type
 
 from cudf._lib.utils cimport columns_from_pylibcudf_table, data_from_pylibcudf_table
 import pylibcudf as plc
-from libc.stdint cimport uintptr_t
-from rmm.pylibrmm.device_buffer cimport DeviceBuffer
 
 
 def _gather_map_is_valid(
@@ -348,10 +346,9 @@ class PackedColumns(Serializable):
     def serialize(self):
         header = {}
         frames = []
-        cdef DeviceBuffer dbuf = self._gpu_data.obj
         gpu_data = as_buffer(
-            data = int(<uintptr_t>dbuf.c_obj.get()[0].data()),
-            size = int(<uintptr_t>dbuf.c_obj.get()[0].size()),
+            data = self._gpu_data.obj.ptr,
+            size = self._gpu_data.obj.size,
             owner=self,
             exposed=True
         )