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!";
|
MS_LOG(INFO) << "Start!";
|
||||||
auto ms_context = MsContext::GetInstance();
|
auto ms_context = MsContext::GetInstance();
|
||||||
MS_EXCEPTION_IF_NULL(ms_context);
|
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!";
|
MS_LOG(INFO) << "Finish!";
|
||||||
return;
|
return;
|
||||||
}
|
}
|
||||||
|
|
Loading…
Reference in New Issue