From a91f076e672d6d56ff6a373e33d250b95f9f530f Mon Sep 17 00:00:00 2001
From: lvchangquan <lvchangquan@huawei.com>
Date: Tue, 21 Jul 2020 16:59:58 +0800
Subject: [PATCH]   fix bug with use trans_data to reduce print time in graph
 mode

---
 .../ccsrc/runtime/device/ascend/ascend_device_address.cc      | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/mindspore/ccsrc/runtime/device/ascend/ascend_device_address.cc b/mindspore/ccsrc/runtime/device/ascend/ascend_device_address.cc
index 26dbef12e4..a4eb11adba 100644
--- a/mindspore/ccsrc/runtime/device/ascend/ascend_device_address.cc
+++ b/mindspore/ccsrc/runtime/device/ascend/ascend_device_address.cc
@@ -482,7 +482,9 @@ bool AscendDeviceAddress::SyncDeviceToHostAndConvertFormat(const std::vector<int
     host_shape.emplace_back(1);
   }
   std::vector<size_t> device_shape = GetDeviceShape(&host_shape);
-  if (type_id_name_map.find(type_id_) != type_id_name_map.end()) {
+  auto ms_context = MsContext::GetInstance();
+  MS_EXCEPTION_IF_NULL(ms_context);
+  if (ms_context->execution_mode() == kPynativeMode && type_id_name_map.find(type_id_) != type_id_name_map.end()) {
     std::pair<std::string, std::string> type_format = std::make_pair(type_id_name_map.at(type_id_), format_);
     if (use_trans_data.find(type_format) != use_trans_data.end()) {
       sync_ok = SyncDeviceToHostAndConvertFormatBasedOnTransData(host_shape, device_shape, size, type, host_ptr);