From cda400b42bb47ad4fb18e7d7eb0dabb2c867632e Mon Sep 17 00:00:00 2001 From: duanjunwen <935724073@qq.com> Date: Thu, 13 Jun 2024 08:12:07 +0000 Subject: [PATCH] [fix] fix test_checkpoint_io/test_hybrid parallel plugin checkpoint io.py --- colossalai/pipeline/p2p.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/colossalai/pipeline/p2p.py b/colossalai/pipeline/p2p.py index 1b55b140c0ba..6befa793d740 100644 --- a/colossalai/pipeline/p2p.py +++ b/colossalai/pipeline/p2p.py @@ -266,7 +266,9 @@ def _send_recv_serialization_object( send_object_tensor = None if object is not None and send_dst is not None: if Version(torch.__version__) >= Version("1.13.0"): - send_object_tensor, send_object_size_tensor = c10d._object_to_tensor(object, device=current_device) + send_object_tensor, send_object_size_tensor = c10d._object_to_tensor( + object, device=current_device, group=c10d._get_default_group() + ) else: send_object_tensor, send_object_size_tensor = c10d._object_to_tensor(object)