Skip to content

Commit 999d0fd

Browse files
committed
By default is all reduce
1 parent c8f3ed2 commit 999d0fd

File tree

2 files changed

+3
-13
lines changed

2 files changed

+3
-13
lines changed

paddle/fluid/framework/details/build_strategy.h

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -27,7 +27,7 @@ struct BuildStrategy {
2727
kCustomized = 2,
2828
};
2929

30-
ReduceStrategy reduce_{ReduceStrategy::kReduce};
30+
ReduceStrategy reduce_{ReduceStrategy::kAllReduce};
3131
GradientScaleStrategy gradient_scale_{GradientScaleStrategy::kCoeffNumDevice};
3232
};
3333

paddle/fluid/framework/details/fetch_op_handle.cc

Lines changed: 2 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -44,12 +44,6 @@ void FetchOpHandle::WaitAndMergeCPUTensors() const {
4444
data_->at(offset_).MergeLoDTensor(tensors_ptr, platform::CPUPlace());
4545
}
4646

47-
template <typename T, typename... ARGS>
48-
inline T &Ref(T *ptr, ARGS &&... args) {
49-
PADDLE_ENFORCE(ptr != nullptr, args...);
50-
return *ptr;
51-
}
52-
5347
void FetchOpHandle::RunImpl() {
5448
WaitInputVarGenerated(platform::CPUPlace());
5549

@@ -59,14 +53,10 @@ void FetchOpHandle::RunImpl() {
5953
platform::CPUPlace cpu;
6054
auto &scopes = *local_scopes_;
6155

62-
PADDLE_ENFORCE_EQ(inputs_.size(), scopes.size());
6356
for (size_t i = 0; i < scopes.size(); ++i) {
6457
auto &scope = scopes[i];
65-
auto *var = Ref(Ref(scope->FindVar(kLocalExecScopeName), "Cannot find %s",
66-
kLocalExecScopeName)
67-
.Get<Scope *>(),
68-
"Cannot get scope")
69-
.FindVar(var_name);
58+
auto *var =
59+
scope->FindVar(kLocalExecScopeName)->Get<Scope *>()->FindVar(var_name);
7060
PADDLE_ENFORCE_NOT_NULL(var, "Cannot find variable %s in execution scope",
7161
var_name);
7262
auto &t = var->Get<framework::LoDTensor>();

0 commit comments

Comments
 (0)