[Semi-Auto] Move and adapt elementwise rule to phi #57197
Conversation
…o semi-auto/default-rule
|
你的PR提交成功，感谢你对开源项目的贡献! |
pkuzyc
force-pushed
the
adapt_phi/elementwise
branch
from
00ac565 to
cd0b2aa
Compare
| @@ -1,213 +0,0 @@ | |||
| /* Copyright (c) 2023 PaddlePaddle Authors. All Rights Reserved. | |||
There was a problem hiding this comment.
not rm, use git mv instead so that git maintain the update and commits from very beginning.
There was a problem hiding this comment.
git automatically modified it to "new" and "delete" when the files change a lot.
| def setUp(self): | ||
| self.rule = get_spmd_rule("add") | ||
| self.unary_rule = core.get_phi_spmd_rule("relu") | ||
| self.rule = core.get_phi_spmd_rule("add") |
There was a problem hiding this comment.
should add test for binary case.
There was a problem hiding this comment.
Done, the binary cases are already in unit tests. Now rename 'self.rule' to 'self.binary_rule' to make it clearer.
| SpmdInfo ElementwiseBinaryInferSpmdReverse(const DistMetaTensor& x, | ||
| const DistMetaTensor& y, | ||
| const DistMetaTensor& out); | ||
|
|
There was a problem hiding this comment.
add rule for more 3 input case (sum, addn)
There was a problem hiding this comment.
Marked a TODO in rules.h
pkuzyc
force-pushed
the
adapt_phi/elementwise
branch
from
cd0b2aa to
e93b129
Compare
* adapt general spmd rule * polish details * add new rules * bugfix --------- Co-authored-by: Chen Weihang <chenweihang@baidu.com> Co-authored-by: liangjianzhong <liangjianzhong@baidu.com>
* adapt general spmd rule * polish details * add new rules * bugfix --------- Co-authored-by: Chen Weihang <chenweihang@baidu.com> Co-authored-by: liangjianzhong <liangjianzhong@baidu.com>
3 participants
PR types
Function optimization
PR changes
Others
Description
Pcard-70448
Adapt elementwise spmd rule to phi api: paddle/fluid/distributed/auto_parallel/spmd_rules/elementwise.* -> paddle/phi/infermeta/spmd_rules/elementwise.*
To adapt phi api, the elementwise is implemented with unary and binary version.