From 6a0483bf70dd73151734fe7643e79a4a74ca5ef4 Mon Sep 17 00:00:00 2001 From: galipremsagar Date: Wed, 11 Dec 2024 16:46:55 +0000 Subject: [PATCH] update benchmarks --- ucp/benchmarks/cudf_merge.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/ucp/benchmarks/cudf_merge.py b/ucp/benchmarks/cudf_merge.py index 07af7b278..8850eb886 100644 --- a/ucp/benchmarks/cudf_merge.py +++ b/ucp/benchmarks/cudf_merge.py @@ -72,8 +72,7 @@ async def recv_df(ep): for frame in frames: await ep.recv(frame) - cudf_typ = pickle.loads(header["type-serialized"]) - return cudf_typ.deserialize(header, frames) + return cudf.Serializable.device_deserialize(header, frames) async def barrier(rank, eps):