!3081 Enable stream sync when context is graph but run pynative

Merge pull request !3081 from JoyLvliang/enable-stream-sync-when-context-is-graph-but-run-in-pynative
5 years ago · 2c3f02b653
parent 220f090ce5 b48592ce79
commit 2c3f02b653
1 changed files with 1 additions and 1 deletions
--- 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;
  }