From 861546921c1688a64dc4591c9fd610aae4e2e2eb Mon Sep 17 00:00:00 2001
From: jinyaohui <jinyaohui@huawei.com>
Date: Sun, 28 Jun 2020 14:31:11 +0800
Subject: [PATCH] fix bug

---
 mindspore/ccsrc/utils/tensorprint_utils.cc | 15 +++++++++------
 mindspore/context.py                       |  2 +-
 2 files changed, 10 insertions(+), 7 deletions(-)

diff --git a/mindspore/ccsrc/utils/tensorprint_utils.cc b/mindspore/ccsrc/utils/tensorprint_utils.cc
index cf7b339edd..ee53345f31 100644
--- a/mindspore/ccsrc/utils/tensorprint_utils.cc
+++ b/mindspore/ccsrc/utils/tensorprint_utils.cc
@@ -214,10 +214,10 @@ bool ConvertDataItem2Tensor(const std::vector<tdt::DataItem> &items) {
 
 bool SaveDataItem2File(const std::vector<tdt::DataItem> &items, const std::string &print_file_path, prntpb::Print print,
                        std::fstream *output) {
-  bool ret_end_sequence = false;
+  bool ret_end_thread = false;
   for (auto &item : items) {
     if (item.dataType_ == tdt::TDT_END_OF_SEQUENCE) {
-      ret_end_sequence = true;
+      ret_end_thread = true;
       break;
     }
     prntpb::Print_Value *value = print.add_value();
@@ -225,14 +225,16 @@ bool SaveDataItem2File(const std::vector<tdt::DataItem> &items, const std::strin
     MS_EXCEPTION_IF_NULL(str_data_ptr);
     if (item.tensorShape_ == kShapeScalar || item.tensorShape_ == kShapeNone) {
       if (!judgeLengthValid(str_data_ptr->size(), item.tensorType_)) {
-        MS_LOG(EXCEPTION) << "Print op receive data length is invalid.";
+        MS_LOG(ERROR) << "Print op receive data length is invalid.";
+        ret_end_thread = true;
       }
     }
 
     std::vector<int> tensor_shape;
     size_t totaldims = 1;
     if (!ParseTensorShape(item.tensorShape_, &tensor_shape, &totaldims)) {
-      MS_LOG(EXCEPTION) << "Tensor print can not parse tensor shape, receive info" << item.tensorShape_;
+      MS_LOG(ERROR) << "Tensor print can not parse tensor shape, receive info" << item.tensorShape_;
+      ret_end_thread = true;
     }
 
     if (item.tensorType_ == "string") {
@@ -252,11 +254,12 @@ bool SaveDataItem2File(const std::vector<tdt::DataItem> &items, const std::strin
     }
 
     if (!print.SerializeToOstream(output)) {
-      MS_LOG(EXCEPTION) << "Save print file:" << print_file_path << " fail.";
+      MS_LOG(ERROR) << "Save print file:" << print_file_path << " fail.";
+      ret_end_thread = true;
     }
     print.Clear();
   }
-  return ret_end_sequence;
+  return ret_end_thread;
 }
 
 void TensorPrint::operator()() {
diff --git a/mindspore/context.py b/mindspore/context.py
index 257ed20766..a355112922 100644
--- a/mindspore/context.py
+++ b/mindspore/context.py
@@ -564,7 +564,7 @@ def set_context(**kwargs):
         max_device_memory (str): Sets the maximum memory available for device, currently only supported on GPU.
             The format is "xxGB". Default: "1024GB".
         print_file_path (str): The path of print data to save. If this parameter is set, print data is saved to
-        a file by default，and turn off printing to the screen.
+            a file by default, and turn off printing to the screen.
 
     Raises:
         ValueError: If input key is not an attribute in context.