switch the flag

7 years ago · 23a674c98a
parent defd7ec641
commit 23a674c98a
1 changed files with 2 additions and 2 deletions
--- 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) {