follow comments

paddle/framework/block_desc.cc

@@ -90,7 +90,7 @@ OpDesc *BlockDesc::PrependOp() {
   return ops_.front().get();
 }
 
-void BlockDescBind::RemoveOp(size_t s, size_t e) {
+void BlockDesc::RemoveOp(size_t s, size_t e) {
   if (ops_.begin() + s == ops_.end() || ops_.begin() + e == ops_.end()) {
     return;
   }

paddle/operators/detail/recv_impl.cc

@@ -58,7 +58,7 @@ Status SendRecvServerImpl::Wait(ServerContext *context,
   return Status::OK;
 }
 
-void SendRecvServerImpl::Start() {
+void SendRecvServerImpl::Reset() {
   std::lock_guard<std::mutex> lock(this->mutex_);
   done_ = false;
 }

paddle/operators/detail/send_recv_impl.h

@@ -56,7 +56,7 @@ class SendRecvServerImpl final : public SendRecvService::Service {
                      VariableMessage *out_var) override;
   Status Wait(ServerContext *context, const VoidMessage *in_var,
               VoidMessage *out_var) override;
-  void Start();
+  void Reset();
   void Done();
   void SetScope(framework::Scope *scope) { scope_ = scope; };
 

paddle/operators/recv_op.cc

@@ -80,7 +80,7 @@ class RecvOp : public framework::OperatorBase {
     auto grad_list = Attr<std::vector<std::string>>("GradList");
     auto trainer_count = Attr<int>("Trainers");
     size_t param_count = param_list.size();
-    rpc_service_->Start();
+    rpc_service_->Reset();
     // TODO(typhoonzero): change this to a while_op for every cluster-batch.
     while (true) {
       // Get from multiple trainers, we don't care about order in which
@@ -93,6 +93,8 @@ class RecvOp : public framework::OperatorBase {
         std::string param_var_name;
         if (it != grad_list.end()) {
           param_var_name = param_list[it - grad_list.begin()];
+        } else {
+          LOG(ERROR) << "grad have no paired param found!";
         }
         VLOG(3) << "recved grad: " << grad_var_name
                 << " updating param: " << param_var_name;
@@ -112,7 +114,7 @@ class RecvOp : public framework::OperatorBase {
         // FIXME(typhoonzero): do not copy
         framework::CopyFrom(v.second, dev_ctx.GetPlace(), dev_ctx, tensor);
       }
-      rpc_service_->Start();
+      rpc_service_->Reset();
 
       std::string program_str = Attr<std::string>("OptimizeProgram");
       framework::ProgramDesc program_desc;