!166 Enable auto-mixed-precision in GeInitialize

Merge pull request !166 from chenhaozhe/pass-auto-mixed-precision-flag
pull/166/MERGE
mindspore-ci-bot 5 years ago committed by Gitee
commit 606310d9c3

@ -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
} }

Loading…
Cancel
Save