solve bug in pull_dense_worker (#27918)

* op error info * style * code format * create pin var bug
5 years ago · 3ee6ad6ec5
parent 263a9e97fd
commit 3ee6ad6ec5
1 changed files with 1 additions and 1 deletions
--- a/paddle/fluid/framework/pull_dense_worker.cc
+++ b/paddle/fluid/framework/pull_dense_worker.cc
@ -67,7 +67,7 @@ void PullDenseWorker::Initialize(const TrainerDesc& param) {
 }

 void PullDenseWorker::CreatePinVar() {
-#if (defined PADDLE_WITH_CUDA) || (defined PADDLE_WITH_PSLIB)
+#if (defined PADDLE_WITH_CUDA) || (defined PADDLE_WITH_XPU)
  // for (auto& v : dense_value_names_) {
  //  for (auto& name : v.second) {
  for (int i = 0; i < dwp_param_.program_config(0).pull_dense_table_id_size();