mkldnn only need one trainer

tensor-tang · tensor-tang · commit defd7ec6412e · 2017-11-16T23:35:01.000+08:00
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_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() ||

Original file line number	Diff line number	Diff line change
`@@ -137,6 +137,10 @@ void Trainer::init(const std::shared_ptr<TrainerConfigHelper>& config,`
`137`	`137`	`}`
`138`	`138`	`}`
`139`	`139`
	`140`	`+ if (FLAGS_trainer_count > 1) {`
	`141`	`+ CHECK(!FLAGS_use_mkldnn) << "MKLDNN only need 1 trainer";`
	`142`	`+ }`
	`143`	`+`
`140`	`144`	`if (testing) {`
`141`	`145`	`LOG(INFO) << "trainer: in testing mode";`
`142`	`146`	`if (config_->getOptConfig().use_sparse_remote_updater() \|\|`