From 9e50b0be5c468bc77d1dfd31fcc6c89d0a6564d5 Mon Sep 17 00:00:00 2001 From: yangzhenzhang Date: Tue, 2 Feb 2021 14:58:53 +0800 Subject: [PATCH] fix bug for prelu mirror --- .../frontend/parallel/ops_info/prelu_info.cc | 20 ------------------- .../frontend/parallel/ops_info/prelu_info.h | 1 - 2 files changed, 21 deletions(-) diff --git a/mindspore/ccsrc/frontend/parallel/ops_info/prelu_info.cc b/mindspore/ccsrc/frontend/parallel/ops_info/prelu_info.cc index 0a484d340ee..dc6ee08e275 100644 --- a/mindspore/ccsrc/frontend/parallel/ops_info/prelu_info.cc +++ b/mindspore/ccsrc/frontend/parallel/ops_info/prelu_info.cc @@ -61,26 +61,6 @@ Status PReLUInfo::InferDevMatrixShape() { return SUCCESS; } -Status PReLUInfo::InferMirrorOps() { - Shape param_tensor_map = inputs_tensor_map_[1]; - std::vector param_group; - if (CreateGroupByTensorMap(param_tensor_map, ¶m_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; } /* diff --git a/mindspore/ccsrc/frontend/parallel/ops_info/prelu_info.h b/mindspore/ccsrc/frontend/parallel/ops_info/prelu_info.h index 401afb3c963..e1990772e8f 100644 --- a/mindspore/ccsrc/frontend/parallel/ops_info/prelu_info.h +++ b/mindspore/ccsrc/frontend/parallel/ops_info/prelu_info.h @@ -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;