-
Notifications
You must be signed in to change notification settings - Fork 5.6k
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
[CustomDevice] GetCCLComm add custom device support #47168
Conversation
21a2f80
to
77e9d11
Compare
77e9d11
to
35e3bc8
Compare
73e94ea
to
6607c72
Compare
#include "paddle/fluid/distributed/collective/ProcessGroupCustom.h" | ||
#endif | ||
|
||
namespace phi { |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
在distributed下加phi的namespace这个操作有点奇怪,既然单独出来了,不能用distributed的namespace吗
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
这个接口在 phi/kernel/sync_batch_norm_kernel.h以phi命名空间导出的,写到distributed对用户而言有点奇怪
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
如果反过来,让phi/kernel/sync_batch_norm_kernel.h里调用collective下的这个方法呢
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
collective的方法对外都是隐藏的,现在 GetCCLComm定义在 phi/kernel/sync_batch_norm_kernel.h,实现在 paddle/fluid/distributed/collective/CommUtils.cc
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
那实现可以先统一放在phi下吗
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
更新了
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
LGTM
@@ -0,0 +1,65 @@ | |||
// Copyright (c) 2022 PaddlePaddle Authors. All Rights Reserved. |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
这个文件也不适合放到phi/kernels目录下,建议移到backends目录下
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
更新了
PR types
Others
PR changes
Others
Describe
GetCCLComm add custom device support