|
|
|
@ -34,7 +34,7 @@ MultiDevSSAGraphBuilder::MultiDevSSAGraphBuilder(
|
|
|
|
|
const std::vector<platform::Place> &places,
|
|
|
|
|
const std::string &loss_var_name,
|
|
|
|
|
const std::unordered_set<std::string> ¶ms,
|
|
|
|
|
const std::vector<Scope *> &local_scopes, bool skip_scale_loss,
|
|
|
|
|
const std::vector<Scope *> &local_scopes, bool use_default_grad_scale,
|
|
|
|
|
platform::NCCLContextMap *nccl_ctxs)
|
|
|
|
|
: loss_var_name_(loss_var_name),
|
|
|
|
|
places_(places),
|
|
|
|
|