File tree Expand file tree Collapse file tree 2 files changed +3
-13
lines changed
paddle/fluid/framework/details Expand file tree Collapse file tree 2 files changed +3
-13
lines changed Original file line number Diff line number Diff line change @@ -27,7 +27,7 @@ struct BuildStrategy {
27
27
kCustomized = 2 ,
28
28
};
29
29
30
- ReduceStrategy reduce_{ReduceStrategy::kReduce };
30
+ ReduceStrategy reduce_{ReduceStrategy::kAllReduce };
31
31
GradientScaleStrategy gradient_scale_{GradientScaleStrategy::kCoeffNumDevice };
32
32
};
33
33
Original file line number Diff line number Diff line change @@ -44,12 +44,6 @@ void FetchOpHandle::WaitAndMergeCPUTensors() const {
44
44
data_->at (offset_).MergeLoDTensor (tensors_ptr, platform::CPUPlace ());
45
45
}
46
46
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
-
53
47
void FetchOpHandle::RunImpl () {
54
48
WaitInputVarGenerated (platform::CPUPlace ());
55
49
@@ -59,14 +53,10 @@ void FetchOpHandle::RunImpl() {
59
53
platform::CPUPlace cpu;
60
54
auto &scopes = *local_scopes_;
61
55
62
- PADDLE_ENFORCE_EQ (inputs_.size (), scopes.size ());
63
56
for (size_t i = 0 ; i < scopes.size (); ++i) {
64
57
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);
70
60
PADDLE_ENFORCE_NOT_NULL (var, " Cannot find variable %s in execution scope" ,
71
61
var_name);
72
62
auto &t = var->Get <framework::LoDTensor>();
You can’t perform that action at this time.
0 commit comments