xyy1
/
mindspore2022
forked from huawei/mindspore2022
0
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

code check clean

This commit is contained in:
wuweikang 2022-03-23 12:48:08 +08:00
parent 20006fcf39
commit 942f20dd2a
2 changed files with 5 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
mindspore/ccsrc/backend/common/pass/reduce_sum_optimizer.cc
View File

@ -141,7 +141,7 @@ AnfNodePtr ReduceSumOptimizer::NewAssistValueNode(const CNodePtr &cnode, const K
for (auto &iter : value_tuple->value()) {
auto item = GetValue<int64_t>(iter->cast<ScalarPtr>());
if (item < 0) {
(void)axes_value.emplace_back(item + x_shape->shape().size());
(void)axes_value.emplace_back(item + static_cast<int64_t>(x_shape->shape().size()));
} else {
(void)axes_value.emplace_back(item);
}

7
mindspore/ccsrc/plugin/device/ascend/kernel/tbe/tbe_json/fusion_tbe_json_creator.cc
View File

@ -293,9 +293,10 @@ void FusionBuildTbeJsonCreator::GenReusedOutputDesc(const AnfNodePtr &anf_node,
std::vector<size_t> FusionBuildTbeJsonCreator::GetDescOutputIndex(const std::vector<int64_t> &output_used_nums) {
std::vector<size_t> desc_output_index = {};
for (size_t idx = 0; idx < output_used_nums.size(); ++idx) {
(void)desc_output_index.emplace_back(idx);
if (output_used_nums[idx] > 1) {
desc_output_index.emplace_back(idx);
auto index = idx;
(void)desc_output_index.emplace_back(index);
if (output_used_nums[index] > 1) {
desc_output_index.emplace_back(index);
}
}
return desc_output_index;