mkldnn only need one trainer

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

Showing with 4 addition and 0 deletion

paddle/trainer/Trainer.cpp paddle/trainer/Trainer.cpp +4 -0

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