diff --git a/mindspore/ccsrc/runtime/device/ascend/ascend_device_address.cc b/mindspore/ccsrc/runtime/device/ascend/ascend_device_address.cc index bfd8de81b07..8494656d3d4 100644 --- a/mindspore/ccsrc/runtime/device/ascend/ascend_device_address.cc +++ b/mindspore/ccsrc/runtime/device/ascend/ascend_device_address.cc @@ -477,7 +477,8 @@ bool AscendDeviceAddress::SyncDeviceToHostAndConvertFormat(const std::vector device_shape = GetDeviceShape(&host_shape); 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()) { + if (ms_context->execution_mode() != kGraphMode && ms_context->execution_mode() != kPynativeMode && + type_id_name_map.find(type_id_) != type_id_name_map.end()) { std::pair 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);