From 453f3cff16f12e7d7570117fc6bf9b2c113d4d6f Mon Sep 17 00:00:00 2001 From: galipremsagar Date: Wed, 11 Dec 2024 17:28:54 +0000 Subject: [PATCH] update import --- ucp/benchmarks/cudf_merge.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/ucp/benchmarks/cudf_merge.py b/ucp/benchmarks/cudf_merge.py index 8850eb88..78164da5 100644 --- a/ucp/benchmarks/cudf_merge.py +++ b/ucp/benchmarks/cudf_merge.py @@ -39,7 +39,7 @@ import cudf # noqa: E402 import rmm # noqa: E402 from rmm.allocators.cupy import rmm_cupy_allocator # noqa: E402 - +from cudf.core.abc import Serializable def sizeof_cudf_dataframe(df): return int( @@ -72,7 +72,7 @@ async def recv_df(ep): for frame in frames: await ep.recv(frame) - return cudf.Serializable.device_deserialize(header, frames) + return Serializable.device_deserialize(header, frames) async def barrier(rank, eps):