forked from mindspore-Ecosystem/mindspore
!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
This commit is contained in:
commit
2c3f02b653
|
@ -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;
|
||||
}
|
||||
|
|
Loading…
Reference in New Issue