update benchmarks

rapidsai · Dec 11, 2024 · 6a0483b · 6a0483b
1 parent 38af753
commit 6a0483b
Showing 1 changed file with 1 addition and 2 deletions.
diff --git 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):