[Auto Parallel] Add spmd rule No.5 for mean_all and mean_all_grad ops. #72479
Conversation
ooooo-create requested review from ForFishes, LiYuRio and zhiqiu as code owners 
| 你的PR提交成功，感谢你对开源项目的贡献! |
| @Yeenyeong 可以review |
luotao1 changed the title 
| self.x_dist_tensor_spec.set_dims_mapping([0, -1]) | ||
| result_dist_attrs = self.rule.infer_forward(self.x_dist_tensor_spec) | ||
| | ||
| inferred_input_dist_attrs = result_dist_attrs[0] |
There was a problem hiding this comment.
Before a container is accessed, its length should be checked first.
| test_fused_dropout_add_rule) | ||
| py_test_modules(test_logsumexp_rule MODULES test_logsumexp_rule) | ||
| py_test_modules(test_nonzero_rule MODULES test_nonzero_rule) | ||
| py_test_modules(test_mean_all_rule MODULES test_mean_all_rule) |
There was a problem hiding this comment.
To pass PR-CI-Hygon-DCU-Test, this line should be moved into the "if(NOT WITH_ROCM)" block which appears in next line.
There was a problem hiding this comment.
Thanks! I have modified it in previous commit, which may not be updated yet for you ~
| /re-run approval |
| @ooooo-create 请解决下冲突 |
… add_spmd_mean_all
Codecov ReportAll modified and coverable lines are covered by tests ✅
Additional details and impacted files@@ Coverage Diff @@ ## develop #72479 +/- ## =========================================== Coverage ? 100.00% =========================================== Files ? 1 Lines ? 7 Branches ? 0 =========================================== Hits ? 7 Misses ? 0 Partials ? 0 ☔ View full report in Codecov by Sentry. 🚀 New features to boost your workflow:
|
luotao1 changed the title 
PaddlePaddle#72479) * add spmd rule for pd_op.mean_all * Fix test config for shape and dims_mapping * refine and add cpp tests * reuse ReductionInfer to MeanAll
luotao1 added HappyOpenSource Pro 6 participants
PR Category
Auto Parallel
PR Types
New features
Description
复用了 ReductionInferSpmdBase, ReductionGradInferSpmd，属于 mean 的一种特殊情况