switch the flag

tensor-tang · tensor-tang · commit 23a674c98aee · 2017-11-17T10:33:19.000+08:00
diff --git a/paddle/trainer/Trainer.cpp b/paddle/trainer/Trainer.cpp
@@ -137,8 +137,8 @@ void Trainer::init(const std::shared_ptr<TrainerConfigHelper>& config,
     }
   }
 
-  if (FLAGS_trainer_count > 1) {
-    CHECK(!FLAGS_use_mkldnn) << "MKLDNN only need 1 trainer";
+  if (FLAGS_use_mkldnn) {
+    CHECK_EQ(FLAGS_trainer_count, 1UL) << "MKLDNN only need 1 trainer";
   }
 
   if (testing) {

Original file line number	Diff line number	Diff line change
`@@ -137,8 +137,8 @@ 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";`
	`140`	`+ if (FLAGS_use_mkldnn) {`
	`141`	`+ CHECK_EQ(FLAGS_trainer_count, 1UL) << "MKLDNN only need 1 trainer";`
`142`	`142`	`}`
`143`	`143`
`144`	`144`	`if (testing) {`