提交 2c3f02b6 编写于 作者: M mindspore-ci-bot 提交者: Gitee

!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
......@@ -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;
}
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册