From 43c12b3e2f83f91993ff3e02b75cc4c160cd3d6a Mon Sep 17 00:00:00 2001 From: galipremsagar Date: Wed, 11 Dec 2024 19:02:31 +0000 Subject: [PATCH] fix serialize call --- ucp/benchmarks/cudf_merge.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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)