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

!29181 fix stride slice parallel mask 

Merge pull request !29181 from fangzehua/fix_strided_mask_master
This commit is contained in:
i-robot 2022-01-17 10:48:44 +00:00 committed by Gitee
parent 3b011e6895 18fe447fb9
commit ced581ce7b
No known key found for this signature in database
GPG Key ID: 173E9B9CA92EEF8F
1 changed files with 6 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
mindspore/ccsrc/frontend/parallel/ops_info/strided_slice_info.cc
View File

@ -42,6 +42,11 @@ Status StridedSliceInfo::GetMask(const std::string &mask_name, int64_t *mask_val
return FAILED;
}
}
if (*mask_value != 0) {
MS_LOG(INFO) << "For StridedSlice op attr:" << mask_name << ", the value is " << *mask_value;
}
return SUCCESS;
}
@ -78,8 +83,7 @@ Status StridedSliceInfo::GetAttrs() {
(GetMask(SHRINK_AXIS_MASK, &shrink_axis_mask_) != SUCCESS)) {
return FAILED;
}
has_mask_ = ((begin_mask_ != 0) || (end_mask_ != 0) || (ellipsis_mask_ != 0) || (new_axis_mask_ != 0) ||
(shrink_axis_mask_ != 0));
has_mask_ = ((ellipsis_mask_ != 0) || (new_axis_mask_ != 0) || (shrink_axis_mask_ != 0));
if (input_value_.size() != STRIDED_SLICE_INPUTS_SIZE) {
MS_LOG(ERROR) << name_ << ": The size of input value must be " << STRIDED_SLICE_INPUTS_SIZE << ", but got "