diff --git a/ucp/benchmarks/cudf_merge.py b/ucp/benchmarks/cudf_merge.py index 2b577b93..c30ea352 100644 --- 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)