request_handler_impl.cc 5.2 KB
Newer Older
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22
// Copyright (c) 2018 PaddlePaddle Authors. All Rights Reserved.
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//     http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.

#include <iostream>
#include <string>
#include <vector>

#include "paddle/fluid/framework/data_type.h"
#include "paddle/fluid/framework/lod_tensor.h"
#include "paddle/fluid/framework/scope.h"
#include "paddle/fluid/framework/selected_rows.h"
23 24
#include "paddle/fluid/operators/distributed/request_handler_impl.h"
#include "paddle/fluid/operators/distributed/rpc_server.h"
T
tangwei12 已提交
25
#include "paddle/fluid/string/printf.h"
26 27 28

namespace paddle {
namespace operators {
29
namespace distributed {
30

T
tangwei12 已提交
31 32
// define LOOKUP_TABLE_PATH for checkpoint notify to save lookup table variables
// to directory specified.
T
tangwei12 已提交
33
constexpr char LOOKUP_TABLE_PATH[] = "kLookupTablePath";
T
tangwei12 已提交
34

35 36 37
bool RequestSendHandler::Handle(const std::string& varname,
                                framework::Scope* scope,
                                framework::Variable* invar,
Q
qiaolongfei 已提交
38 39
                                framework::Variable** outvar,
                                const std::string& out_var_name) {
40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57
  VLOG(4) << "RequestSendHandler:" << varname;

  // Async
  if (!sync_mode_) {
    try {
      executor_->RunPreparedContext((*grad_to_prepared_ctx_)[varname].get(),
                                    scope);
    } catch (std::exception& e) {
      LOG(ERROR) << "async: run sub program error " << e.what();
      return false;
    }
    return true;
  }

  // Sync
  if (varname == BATCH_BARRIER_MESSAGE) {
    VLOG(3) << "sync: recv batch barrier message";
    rpc_server_->IncreaseBatchBarrier(kRequestSend);
Y
Yancey1989 已提交
58 59 60 61
  } else if (varname == BEGIN_PASS_MESSAGE) {
    VLOG(3) << "sync: recv begin pass message";
    rpc_server_->WaitCond(kRequestSend);
    rpc_server_->BeginPass();
62 63
  } else {
    VLOG(3) << "sync: received var_name: " << varname;
Y
Yancey1989 已提交
64 65
    rpc_server_->WaitCond(kRequestSend);
    VLOG(3) << "sync: processing received var: " << varname;
66 67 68 69 70 71

    if (invar == nullptr) {
      LOG(ERROR) << "sync: Can not find server side var: " << varname;
      PADDLE_THROW("sync: Can not find server side var");
      return false;
    }
Y
Yancey1989 已提交
72
    if (invar->IsType<framework::SelectedRows>()) {
Y
Yancey1989 已提交
73 74
      std::unique_lock<std::mutex> lock(mutex_sparse_vars_);
      sparse_vars_.push_back(invar);
Y
Yancey1989 已提交
75
    }
76 77 78 79
  }
  return true;
}

Y
Yancey1989 已提交
80 81 82 83 84 85 86 87
void RequestSendHandler::ResetSparseVarRecorder() {
  std::unique_lock<std::mutex> lock(mutex_sparse_vars_);
  for (auto* var : sparse_vars_) {
    var->GetMutable<framework::SelectedRows>()->mutable_rows()->clear();
  }
  sparse_vars_.clear();
}

88 89 90
bool RequestGetHandler::Handle(const std::string& varname,
                               framework::Scope* scope,
                               framework::Variable* invar,
Q
qiaolongfei 已提交
91 92
                               framework::Variable** outvar,
                               const std::string& out_var_name) {
93
  VLOG(4) << "RequestGetHandler:" << varname;
Y
Yancey1989 已提交
94 95 96 97 98 99 100
  if (sync_mode_) {
    if (varname == FETCH_BARRIER_MESSAGE) {
      VLOG(3) << "sync: recv fetch barrier message";
      rpc_server_->IncreaseBatchBarrier(kRequestGet);
    } else if (varname == END_PASS_MESSAGE) {
      rpc_server_->EndPass();
    } else {
101
      rpc_server_->WaitCond(kRequestGet);
Y
Yancey1989 已提交
102 103 104 105 106
      *outvar = scope_->FindVar(varname);
    }
  } else {
    if (varname != FETCH_BARRIER_MESSAGE && varname != END_PASS_MESSAGE) {
      *outvar = scope_->FindVar(varname);
107 108 109 110 111 112 113 114
    }
  }
  return true;
}

bool RequestPrefetchHandler::Handle(const std::string& varname,
                                    framework::Scope* scope,
                                    framework::Variable* invar,
Q
qiaolongfei 已提交
115 116
                                    framework::Variable** outvar,
                                    const std::string& out_var_name) {
117 118
  VLOG(4) << "RequestPrefetchHandler " << varname;

Q
qiaolongfei 已提交
119
  auto var_desc = program_->Block(0).FindVar(out_var_name);
120
  InitializeVariable(*outvar, var_desc->GetType());
121 122
  executor_->RunPreparedContext(
      (*prefetch_var_name_to_prepared_ctx_)[varname].get(), scope);
123 124 125 126

  return true;
}

T
tangwei12 已提交
127 128 129 130
bool RequestCheckpointHandler::Handle(const std::string& varname,
                                      framework::Scope* scope,
                                      framework::Variable* invar,
                                      framework::Variable** outvar,
T
bug fix  
tangwei12 已提交
131
                                      const std::string& out_var_name) {
132 133 134
  PADDLE_ENFORCE(
      checkpoint_notify_id != -1,
      "when checkpoint_notify_id = -1, there should be no RPC invoke.");
T
tangwei12 已提交
135

T
tangwei12 已提交
136
  auto* lt_var = scope->FindVar(LOOKUP_TABLE_PATH)->GetMutable<std::string>();
T
tangwei12 已提交
137 138
  lt_var->clear();
  lt_var->append(out_var_name);
T
tangwei12 已提交
139
  VLOG(4) << "RequestCheckpointHandler update var kLookupTablePath to: "
140
          << out_var_name;
T
bug fix  
tangwei12 已提交
141
  executor_->RunPreparedContext(checkpoint_prepared_ctx_.get(), scope);
T
bug fix  
tangwei12 已提交
142 143
  return true;
}
T
tangwei12 已提交
144

145
}  // namespace distributed
146 147
}  // namespace operators
}  // namespace paddle