sfqcyy
/
mindspore
forked from OSSInnovation/mindspore
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

!6064 Fix dataset api get nullptr of iterator when occur runtime error 

Merge pull request !6064 from jiangzhiwen/fix/iter_null_ptr
This commit is contained in:
mindspore-ci-bot 2020-09-12 14:40:00 +08:00 committed by Gitee
parent 4bb05529d5 0e9fbf6970
commit ec9a34b3cd
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
mindspore/ccsrc/minddata/dataset/api/iterator.cc
View File

@ -119,9 +119,9 @@ Status Iterator::BuildAndLaunchTree(std::shared_ptr<Dataset> ds) {
// Launch the execution tree.
RETURN_IF_NOT_OK(tree_->Prepare());
RETURN_IF_NOT_OK(tree_->Launch());
iterator_ = std::make_unique<DatasetIterator>(tree_);
RETURN_UNEXPECTED_IF_NULL(iterator_);
RETURN_IF_NOT_OK(tree_->Launch());
return rc;
}