Fix CPU compile

wangkuiyi-patch-2
Yu Yang 7 years ago
parent 7a395881d4
commit 55feba9b5a

@ -44,8 +44,8 @@ MultiDevSSAGraphBuilder::MultiDevSSAGraphBuilder(
MultiDevSSAGraphBuilder::MultiDevSSAGraphBuilder( MultiDevSSAGraphBuilder::MultiDevSSAGraphBuilder(
const std::vector<platform::Place> &places, const std::vector<platform::Place> &places,
const std::string &loss_var_name, const std::string &loss_var_name,
const std::unordered_set<std::string> &params, bool skip_scale_loss, const std::unordered_set<std::string> &params,
const std::vector<Scope *> &local_scopes) const std::vector<Scope *> &local_scopes, bool skip_scale_loss)
: loss_var_name_(loss_var_name), : loss_var_name_(loss_var_name),
places_(places), places_(places),
local_scopes_(local_scopes) { local_scopes_(local_scopes) {

@ -40,8 +40,8 @@ class MultiDevSSAGraphBuilder : public SSAGraphBuilder {
MultiDevSSAGraphBuilder(const std::vector<platform::Place> &places, MultiDevSSAGraphBuilder(const std::vector<platform::Place> &places,
const std::string &loss_var_name, const std::string &loss_var_name,
const std::unordered_set<std::string> &params, const std::unordered_set<std::string> &params,
bool skip_scale_loss, const std::vector<Scope *> &local_scopes,
const std::vector<Scope *> &local_scopes); bool skip_scale_loss);
#endif #endif
std::unique_ptr<SSAGraph> Build(const ProgramDesc &program) const override; std::unique_ptr<SSAGraph> Build(const ProgramDesc &program) const override;

Loading…
Cancel
Save