diff --git a/python/paddle/distributed/fleet/meta_parallel/pp_utils/p2p_communication.py b/python/paddle/distributed/fleet/meta_parallel/pp_utils/p2p_communication.py index e97cf3c02302f8a298dbe5f294c5fa0cfcf79dad..7f2e2b8d7a48bd2d1bc412bb53f76d2c3d4b5900 100644 --- a/python/paddle/distributed/fleet/meta_parallel/pp_utils/p2p_communication.py +++ b/python/paddle/distributed/fleet/meta_parallel/pp_utils/p2p_communication.py @@ -164,7 +164,11 @@ class SendRecvMeta: [d.shape for d in tensor if not d.stop_gradient] ) self.send_dtype_message = tuple( - [paddle_2_number(d.dtype) for d in tensor] + [ + paddle_2_number(d.dtype) + for d in tensor + if not d.stop_gradient + ] )