|
|
@ -358,7 +358,9 @@ void MsContext::GetGeOptions(std::map<std::string, std::string>* ge_options) con
|
|
|
|
MS_LOG(ERROR) << "Set proto lib path failed!";
|
|
|
|
MS_LOG(ERROR) << "Set proto lib path failed!";
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
// Disbale the global variable acc, only enable it whlie adding training graph in pipeline
|
|
|
|
// Enable auto mixed precision according to the context options
|
|
|
|
|
|
|
|
(*ge_options)["ge.exec.auto_mix_precision"] = std::to_string(auto_mixed_precision_flag_);
|
|
|
|
|
|
|
|
// Disable the global variable acc, only enable it whlie adding training graph in pipeline
|
|
|
|
(*ge_options)["ge.exec.variable_acc"] = "0";
|
|
|
|
(*ge_options)["ge.exec.variable_acc"] = "0";
|
|
|
|
#endif
|
|
|
|
#endif
|
|
|
|
}
|
|
|
|
}
|
|
|
|