fix loss scale and task_sink_mode to false at the same time cause segmentation fault

This commit is contained in:
wanghua 2020-04-30 10:28:46 +08:00
parent 37e3582794
commit fdbef20638
1 changed files with 1 additions and 1 deletions

View File

@ -990,7 +990,7 @@ void AscendSession::MergeGraphExecOrder() {
auto context_ptr = MsContext::GetInstance();
MS_EXCEPTION_IF_NULL(context_ptr);
if (!context_ptr->enable_task_sink()) {
MS_LOG(INFO) << "Control sink network should run with task-sink mode!";
MS_LOG(EXCEPTION) << "Control sink network should run with task-sink mode!";
}
}
// if first graph is common,the final graph has no label,then set the stream of final graph same with the first graph