fix serialize call

rapidsai · Dec 11, 2024 · 43c12b3 · 43c12b3
1 parent 9ca348c
commit 43c12b3
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/ucp/benchmarks/cudf_merge.py b/ucp/benchmarks/cudf_merge.py
@@ -49,7 +49,7 @@ def sizeof_cudf_dataframe(df):
 
 
 async def send_df(ep, df):
-    header, frames = df.serialize()
+    header, frames = df.device_serialize()
     header["frame_ifaces"] = [f.__cuda_array_interface__ for f in frames]
     header = pickle.dumps(header)
     header_nbytes = np.array([len(header)], dtype=np.uint64)