-
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
【PIR Dist Op Reg No.3】 reg c_allreduce_min #60584
Conversation
你的PR提交成功,感谢你对开源项目的贡献! |
可以rerun一下CI,应该可以合入了 |
already done. thanks |
'c_allreduce_min', | ||
'c_allreduce_min_', |
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.
这两处的用途不一样,名单也不完全重合,第二类主要是针对struct kernel,我觉得可能可以优化,之后研究下。
@@ -55,6 +55,7 @@ const std::unordered_set<std::string> LegacyOpList = { | |||
CReduceSumOp::name(), | |||
CReduceSum_Op::name(), | |||
CAllreduceMax_Op::name(), | |||
CAllreduceMin_Op::name(), |
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.
同上
@@ -55,6 +55,7 @@ const std::unordered_set<std::string> LegacyOpList = { | |||
CReduceSumOp::name(), | |||
CReduceSum_Op::name(), | |||
CAllreduceMax_Op::name(), | |||
CAllreduceMin_Op::name(), |
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.
这里为什么没有CAllreduceMinOp::name()
?
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.
@enkilee 这里可以提个补充PR,这个PR我先合入吧
* fix * fix * fix * fix * fix * fix
PR types
Others
PR changes
Others
Description
注册算子
c_allreduce_min