From 0e9fbf697009ae375860f090a7d0258a752fb5cd Mon Sep 17 00:00:00 2001
From: jiangzhiwen <jiangzhiwen8@huawei.com>
Date: Fri, 11 Sep 2020 16:52:22 +0800
Subject: [PATCH] fix iterator get null when occur runtimer error

---
 mindspore/ccsrc/minddata/dataset/api/iterator.cc | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/mindspore/ccsrc/minddata/dataset/api/iterator.cc b/mindspore/ccsrc/minddata/dataset/api/iterator.cc
index df1b8e7b51..d892135147 100644
--- a/mindspore/ccsrc/minddata/dataset/api/iterator.cc
+++ b/mindspore/ccsrc/minddata/dataset/api/iterator.cc
@@ -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;
 }