switch the flag

23a674c9 · tensor-tang · defd7ec6 · 23a674c9
显示空白变更内容
内联并排

Showing with 2 addition and 2 deletion

paddle/trainer/Trainer.cpp paddle/trainer/Trainer.cpp +2 -2

未找到文件。
--- 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) {