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

!26383 reshape_dont_replace_shape_when_no_other_redistribution 

Merge pull request !26383 from yao_yf/reshape_dont_replace_shape_when_no_other_redistribution
This commit is contained in:
i-robot 2021-11-23 01:07:34 +00:00 committed by Gitee
parent 02e53a96b7 94e1d5dfe0
commit 6d4837fe93
1 changed files with 15 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
mindspore/ccsrc/frontend/parallel/ops_info/reshape_info.cc
View File

@ -146,6 +146,21 @@ Status ReshapeInfo::ComputeReplaceOp() {
replace_op_info_ = redistribution_oplist_ptr->second;
}
MS_LOG(DEBUG) << name_ << ": replace op size = " << replace_op_.size();
if (replace_op_.size() == 1 && replace_op_.front().first == RESHAPE) {
int64_t shape_dim = 2;
auto value = replace_op_.front().second.second.front().first.second;
Shape dst_shape = GetValue<std::vector<int64_t>>(value);
Shape origin_dst_shape = GetValue<std::vector<int64_t>>(cnode_->input(shape_dim)->cast<ValueNodePtr>()->value());
if (dst_shape.size() == origin_dst_shape.size()) {
for (size_t i = 0; i < dst_shape.size(); ++i) {
if (origin_dst_shape[i] != dst_shape[i] && origin_dst_shape[i] != -1) {
return SUCCESS;
}
}
MS_LOG(INFO) << "The reshape would not change the target shape.";
replace_op_.front().second.second.front().first.second = MakeValue(origin_dst_shape);
}
}
return SUCCESS;
}