Skip to content

Commit 8d8388c

Browse files
authored
Merge pull request #686 from emailweixu/fix_test
Fix --job=test
2 parents 257819d + 66520f8 commit 8d8388c

File tree

2 files changed

+4
-2
lines changed

2 files changed

+4
-2
lines changed

paddle/gserver/dataproviders/DataProvider.h

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -271,7 +271,9 @@ class DoubleBuffer {
271271
void finishAsyncLoad() {
272272
stopping_ = true;
273273
taskReadySem_.post();
274-
asyncLoader_->join();
274+
if (asyncLoader_) {
275+
asyncLoader_->join();
276+
}
275277
}
276278

277279
void setPending(bool pending) { pending_ = pending; }

paddle/trainer/Trainer.cpp

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -205,7 +205,7 @@ void Trainer::init(const std::shared_ptr<TrainerConfigHelper>& config,
205205
(!IGradientMachineMode::dataMustInCpu(mode_, FLAGS_trainer_count));
206206

207207
dataProvider_ = dataProvider;
208-
if (!dataProvider_ && config_->hasDataConfig()) {
208+
if (!dataProvider_ && config_->hasDataConfig() && !testing_) {
209209
dataProvider_.reset(DataProvider::create(*config_, *config_, gpuData));
210210
}
211211
if (!testDataProvider_) {

0 commit comments

Comments
 (0)