mindspore-Ecosystem
/
mindspore
21
0
Fork 27
Code Issues Pull Requests 6 Projects Releases Wiki Activity

Fix the case of negative index in GetitemDependReorder

This commit is contained in:
yujianfeng 2021-03-22 09:14:36 +08:00
parent 3fbeb9d701
commit 3b521a1e18
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
mindspore/ccsrc/frontend/optimizer/irpass/item_tuple_or_list_eliminate.h
View File

@ -370,7 +370,10 @@ class GetitemDependReorder : public AnfVisitor {
int64_t idx = idx_value->value();
if (abs->isa<abstract::AbstractTuple>()) {
auto abs_tuple = abs->cast<abstract::AbstractTuplePtr>();
if (LongToSize(idx) >= abs_tuple->elements().size() || idx < 0) {
if (idx < 0) {
idx += abs_tuple->elements().size();
}
if (idx < 0 || LongToSize(idx) >= abs_tuple->elements().size()) {
MS_LOG(EXCEPTION) << "The idx value " << idx << " of tuple_getitem node " << c_->DebugString()
<< " is out of range.";
}