Merge pull request PaddlePaddle#5715 from tensor-tang/check

mkldnn only need one trainer
QingshuChen · Nov 17, 2017 · ba9f252 · ba9f252
2 parents c808fbb + 23a674c
commit ba9f252
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/paddle/trainer/Trainer.cpp b/paddle/trainer/Trainer.cpp
@@ -137,6 +137,10 @@ void Trainer::init(const std::shared_ptr<TrainerConfigHelper>& config,
     }
   }
 
+  if (FLAGS_use_mkldnn) {
+    CHECK_EQ(FLAGS_trainer_count, 1UL) << "MKLDNN only need 1 trainer";
+  }
+
   if (testing) {
     LOG(INFO) << "trainer: in testing mode";
     if (config_->getOptConfig().use_sparse_remote_updater() ||