mkldnn only need one trainer

7 years ago · defd7ec641
parent bf712b5cef
commit defd7ec641
1 changed files with 4 additions and 0 deletions
--- 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_trainer_count > 1) {
+    CHECK(!FLAGS_use_mkldnn) << "MKLDNN only need 1 trainer";
+  }
+
  if (testing) {
    LOG(INFO) << "trainer: in testing mode";
    if (config_->getOptConfig().use_sparse_remote_updater() ||