提交 55feba9b 编写于 作者: Y Yu Yang

Fix CPU compile

上级 7a395881
...@@ -44,8 +44,8 @@ MultiDevSSAGraphBuilder::MultiDevSSAGraphBuilder( ...@@ -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 { ...@@ -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;
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册