You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
90 lines
2.4 KiB
90 lines
2.4 KiB
/* Copyright (c) 2019 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. */
|
|
|
|
#pragma once
|
|
|
|
#include <iostream>
|
|
#include <string>
|
|
#include <vector>
|
|
|
|
namespace paddle {
|
|
namespace operators {
|
|
namespace distributed {
|
|
|
|
struct RpcContext {
|
|
RpcContext() = default;
|
|
|
|
RpcContext(const std::string &name, const std::vector<std::string> &names,
|
|
const std::vector<std::string> &emap,
|
|
const std::vector<int64_t> §ions, int id,
|
|
bool merge_add_ = true, bool use_send_handler_ = true)
|
|
: var_name(name),
|
|
splited_var_names(names),
|
|
epmap(emap),
|
|
height_sections(sections),
|
|
trainer_id(id),
|
|
merge_add(merge_add_),
|
|
use_send_handler(use_send_handler_) {}
|
|
|
|
RpcContext(const RpcContext &ctx) {
|
|
var_name = ctx.var_name;
|
|
splited_var_names = ctx.splited_var_names;
|
|
epmap = ctx.epmap;
|
|
height_sections = ctx.height_sections;
|
|
trainer_id = ctx.trainer_id;
|
|
merge_add = ctx.merge_add;
|
|
use_send_handler = ctx.use_send_handler;
|
|
}
|
|
|
|
std::string var_name;
|
|
std::vector<std::string> splited_var_names;
|
|
std::vector<std::string> epmap;
|
|
std::vector<int64_t> height_sections;
|
|
int trainer_id;
|
|
bool merge_add;
|
|
bool use_send_handler;
|
|
};
|
|
|
|
inline std::ostream &operator<<(std::ostream &os, const RpcContext &rpc_ctx) {
|
|
os << "{";
|
|
os << "var_name: " << rpc_ctx.var_name << "\n";
|
|
|
|
os << "splited_var_names: [";
|
|
for (auto &name : rpc_ctx.splited_var_names) {
|
|
os << name << ", ";
|
|
}
|
|
os << "]\n";
|
|
|
|
os << "epmap: [";
|
|
for (auto &ep : rpc_ctx.epmap) {
|
|
os << ep << ", ";
|
|
}
|
|
os << "]\n";
|
|
|
|
os << "height_sections: [";
|
|
for (auto §ion : rpc_ctx.height_sections) {
|
|
os << section << ", ";
|
|
}
|
|
os << "]\n";
|
|
|
|
os << "merge add: " << rpc_ctx.merge_add;
|
|
os << "; send handler: " << rpc_ctx.use_send_handler << "\n";
|
|
os << "}";
|
|
return os;
|
|
}
|
|
|
|
} // namespace distributed
|
|
} // namespace operators
|
|
} // namespace paddle
|