Nigel
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

!11976 fix bug for prelu mirror 

From: @yangzhenzhang
Reviewed-by: @cristoval,@stsuteng
Signed-off-by: @stsuteng
This commit is contained in:
mindspore-ci-bot 2021-02-02 19:58:30 +08:00 committed by Gitee
parent e8fba6787c 9e50b0be5c
commit d02f6f4d10
2 changed files with 0 additions and 21 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
mindspore/ccsrc/frontend/parallel/ops_info/prelu_info.cc
View File

@ -61,26 +61,6 @@ Status PReLUInfo::InferDevMatrixShape() {
return SUCCESS;
}
Status PReLUInfo::InferMirrorOps() {
Shape param_tensor_map = inputs_tensor_map_[1];
std::vector<Group> param_group;
if (CreateGroupByTensorMap(param_tensor_map, &param_group) != SUCCESS) {
return FAILED;
} else if (param_group.empty()) {
MS_LOG(INFO) << name_ << ": The mirror ops is empty.";
return SUCCESS;
}
OperatorVector op_for_param;
op_for_param = CreateMirrorOps(param_group[0].name(), param_group[0].GetDevNum());
// op_for_inputs is empty
OperatorVector op_for_inputs;
mirror_ops_.push_back(op_for_inputs);
mirror_ops_.push_back(op_for_param);
std::string group_name = param_group[0].name();
MS_LOG(INFO) << name_ << ": The mirror ops group is " << group_name;
return SUCCESS;
}
Status PReLUInfo::InferForwardCommunication() { return SUCCESS; }
/*

1
mindspore/ccsrc/frontend/parallel/ops_info/prelu_info.h
View File

@ -45,7 +45,6 @@ class PReLUInfo : public OperatorInfo {
protected:
Status CheckStrategy(const StrategyPtr &strategy) override;
Status InferMirrorOps() override;
Status InferForwardCommunication() override;
Status InferTensorInfo() override;
Status InferDevMatrixShape() override;