From b48592ce79a799d55a6d21b7c41cd2a8bfc1c955 Mon Sep 17 00:00:00 2001
From: lvliang <lvliang18@huawei.com>
Date: Wed, 15 Jul 2020 14:28:32 +0800
Subject: [PATCH] enable-stream-sync-when-context-is-graph-but-run-in-pynative

---
 mindspore/ccsrc/runtime/device/ascend/ascend_device_address.cc | 2 +-
 1 file changed, 1 insertion(+), 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 1a87f3e6af..1b09ef94b6 100644
--- a/mindspore/ccsrc/runtime/device/ascend/ascend_device_address.cc
+++ b/mindspore/ccsrc/runtime/device/ascend/ascend_device_address.cc
@@ -100,7 +100,7 @@ void AscendDeviceAddress::SyncStream() const {
   MS_LOG(INFO) << "Start!";
   auto ms_context = MsContext::GetInstance();
   MS_EXCEPTION_IF_NULL(ms_context);
-  if (ms_context->execution_mode() != kPynativeMode) {
+  if (ms_context->execution_mode() != kPynativeMode && !ms_context->enable_pynative_infer()) {
     MS_LOG(INFO) << "Finish!";
     return;
   }