未验证 提交 cb087beb 编写于 作者: S ShenLiang 提交者: GitHub

fix bug of p2p (#48045)

上级 c20eb7a6
...@@ -164,7 +164,11 @@ class SendRecvMeta: ...@@ -164,7 +164,11 @@ class SendRecvMeta:
[d.shape for d in tensor if not d.stop_gradient] [d.shape for d in tensor if not d.stop_gradient]
) )
self.send_dtype_message = tuple( 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
]
) )
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册