Luo Tao
7 years ago
commit
1992f70920
@ -0,0 +1,128 @@
|
||||
/* Copyright (c) 2016 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 <nccl.h>
|
||||
#include <stdint.h>
|
||||
#include <ostream>
|
||||
#include <string>
|
||||
|
||||
#include "paddle/fluid/framework/executor.h"
|
||||
#include "paddle/fluid/framework/lod_tensor.h"
|
||||
#include "paddle/fluid/framework/op_registry.h"
|
||||
#include "paddle/fluid/framework/threadpool.h"
|
||||
#include "paddle/fluid/operators/detail/grpc_client.h"
|
||||
#include "paddle/fluid/operators/detail/grpc_server.h"
|
||||
#include "paddle/fluid/platform/nccl_helper.h"
|
||||
|
||||
namespace paddle {
|
||||
namespace operators {
|
||||
|
||||
class GenNCCLIdOp : public framework::OperatorBase {
|
||||
public:
|
||||
GenNCCLIdOp(const std::string& type, const framework::VariableNameMap& inputs,
|
||||
const framework::VariableNameMap& outputs,
|
||||
const framework::AttributeMap& attrs)
|
||||
: OperatorBase(type, inputs, outputs, attrs) {}
|
||||
|
||||
void RunImpl(const framework::Scope& scope,
|
||||
const platform::Place& dev_place) const override {
|
||||
platform::DeviceContextPool& pool = platform::DeviceContextPool::Instance();
|
||||
// put nccl id in CPUPlace
|
||||
auto& dev_ctx = *pool.Get(platform::CPUPlace());
|
||||
int trainer_id = Attr<int>("trainer_id");
|
||||
framework::Scope& local_scope = scope.NewScope();
|
||||
|
||||
if (trainer_id == 0) {
|
||||
GenerateAndSend(&local_scope, dev_ctx);
|
||||
} else {
|
||||
GetIdByServer(&local_scope, dev_ctx);
|
||||
}
|
||||
}
|
||||
|
||||
private:
|
||||
void GenerateAndSend(framework::Scope* scope,
|
||||
const platform::DeviceContext& dev_ctx) const {
|
||||
auto var = scope->FindVar(NCCL_ID_VARNAME);
|
||||
PADDLE_ENFORCE_NOT_NULL(var);
|
||||
auto id = var->GetMutable<ncclUniqueId>();
|
||||
PADDLE_ENFORCE(platform::dynload::ncclGetUniqueId(id));
|
||||
|
||||
std::vector<std::string> endpoint_list =
|
||||
Attr<std::vector<std::string>>("endpoint_list");
|
||||
detail::RPCClient client;
|
||||
for (auto& ep : endpoint_list) {
|
||||
VLOG(3) << "sending nccl id to " << ep;
|
||||
client.AsyncSendVariable(ep, dev_ctx, *scope, NCCL_ID_VARNAME);
|
||||
}
|
||||
client.Wait();
|
||||
VLOG(3) << "sending completed...";
|
||||
}
|
||||
|
||||
void GetIdByServer(framework::Scope* scope,
|
||||
const platform::DeviceContext& dev_ctx) const {
|
||||
std::string endpoint = Attr<std::string>("endpoint");
|
||||
// NOTE: Can not use unique_ptr here because the default
|
||||
// deleter will call GRPC Server's base class's dtor and
|
||||
// that will cause a wired crash.
|
||||
detail::AsyncGRPCServer rpc_service(endpoint, true);
|
||||
framework::ProgramDesc empty_program;
|
||||
framework::Executor executor(dev_ctx.GetPlace());
|
||||
rpc_service.SetScope(scope);
|
||||
rpc_service.SetDevCtx(&dev_ctx);
|
||||
rpc_service.SetProgram(&empty_program);
|
||||
rpc_service.SetExecutor(&executor);
|
||||
|
||||
std::thread server_thread(
|
||||
std::bind(&detail::AsyncGRPCServer::RunSyncUpdate, &rpc_service));
|
||||
rpc_service.SetCond(0);
|
||||
VLOG(3) << "start getting nccl id from trainer 0...";
|
||||
auto recv = rpc_service.Get();
|
||||
VLOG(3) << "got nccl id and stop server...";
|
||||
rpc_service.ShutDown();
|
||||
VLOG(3) << "rpc server stopped";
|
||||
server_thread.join();
|
||||
}
|
||||
};
|
||||
|
||||
class GenNCCLIdOpMaker : public framework::OpProtoAndCheckerMaker {
|
||||
public:
|
||||
void Make() override {
|
||||
AddOutput("NCCLID", "Raw variable contains a NCCL UniqueId instaces.");
|
||||
AddComment(R"DOC(
|
||||
GenNCCLId operator
|
||||
|
||||
For trainer 0: generate a new UniqueId and send it to all the other trainers.
|
||||
For trainer 1~n: start a gRPC server to get the UniqueId, once got, stop the server.
|
||||
)DOC");
|
||||
AddAttr<std::string>("endpoint",
|
||||
"(string), e.g. 127.0.0.1:6175 "
|
||||
"current listen endpoint");
|
||||
AddAttr<std::vector<std::string>>(
|
||||
"endpoint_list",
|
||||
"['trainer1_ip:port', 'trainer2_ip:port', ...] "
|
||||
"list of trainer endpoints start from trainer 1")
|
||||
.SetDefault({});
|
||||
AddAttr<int>("trainer_id",
|
||||
"(int default 0) "
|
||||
"The index of the trainer in distributed training.")
|
||||
.SetDefault(0);
|
||||
}
|
||||
};
|
||||
|
||||
} // namespace operators
|
||||
} // namespace paddle
|
||||
|
||||
namespace ops = paddle::operators;
|
||||
|
||||
REGISTER_OPERATOR(gen_nccl_id, ops::GenNCCLIdOp, ops::GenNCCLIdOpMaker);
|
||||
@ -0,0 +1,94 @@
|
||||
/* Copyright (c) 2016 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 <unistd.h>
|
||||
#include <string>
|
||||
#include <thread> // NOLINT
|
||||
|
||||
#include "gtest/gtest.h"
|
||||
#include "paddle/fluid/framework/op_registry.h"
|
||||
#include "paddle/fluid/framework/operator.h"
|
||||
#include "paddle/fluid/framework/program_desc.h"
|
||||
#include "paddle/fluid/operators/detail/grpc_client.h"
|
||||
#include "paddle/fluid/operators/listen_and_serv_op.h"
|
||||
#include "paddle/fluid/operators/math/math_function.h"
|
||||
#include "paddle/fluid/operators/math/selected_rows_functor.h"
|
||||
#include "paddle/fluid/platform/nccl_helper.h"
|
||||
#include "paddle/fluid/string/printf.h"
|
||||
|
||||
USE_NO_KERNEL_OP(listen_and_serv);
|
||||
|
||||
namespace f = paddle::framework;
|
||||
namespace p = paddle::platform;
|
||||
namespace m = paddle::operators::math;
|
||||
namespace detail = paddle::operators::detail;
|
||||
namespace string = paddle::string;
|
||||
|
||||
std::unique_ptr<detail::AsyncGRPCServer> rpc_service;
|
||||
|
||||
void StartServer(std::atomic<bool>* initialized) {
|
||||
f::Scope scope;
|
||||
p::CPUPlace place;
|
||||
scope.Var(NCCL_ID_VARNAME);
|
||||
p::DeviceContextPool& pool = p::DeviceContextPool::Instance();
|
||||
auto& dev_ctx = *pool.Get(p::CPUPlace());
|
||||
|
||||
rpc_service.reset(new detail::AsyncGRPCServer("127.0.0.1:0", true));
|
||||
|
||||
f::ProgramDesc empty_program;
|
||||
f::Executor executor(dev_ctx.GetPlace());
|
||||
rpc_service->SetScope(&scope);
|
||||
rpc_service->SetDevCtx(&dev_ctx);
|
||||
rpc_service->SetProgram(&empty_program);
|
||||
rpc_service->SetExecutor(&executor);
|
||||
|
||||
std::thread server_thread(
|
||||
std::bind(&detail::AsyncGRPCServer::RunSyncUpdate, rpc_service.get()));
|
||||
*initialized = true;
|
||||
rpc_service->SetCond(0);
|
||||
auto recv = rpc_service->Get();
|
||||
LOG(INFO) << "got nccl id and stop server...";
|
||||
rpc_service->ShutDown();
|
||||
server_thread.join();
|
||||
}
|
||||
|
||||
TEST(SendNcclId, Normal) {
|
||||
std::atomic<bool> initialized{false};
|
||||
std::thread server_thread(StartServer, &initialized);
|
||||
while (!initialized) {
|
||||
}
|
||||
// wait server to start
|
||||
// sleep(2);
|
||||
rpc_service->WaitServerReady();
|
||||
|
||||
f::Scope scope;
|
||||
p::CPUPlace place;
|
||||
p::DeviceContextPool& pool = p::DeviceContextPool::Instance();
|
||||
auto& dev_ctx = *pool.Get(p::CPUPlace());
|
||||
|
||||
auto var = scope.Var(NCCL_ID_VARNAME);
|
||||
// var->SetType(f::proto::VarType_Type_RAW);
|
||||
auto id = var->GetMutable<ncclUniqueId>();
|
||||
p::dynload::ncclGetUniqueId(id);
|
||||
|
||||
int port = rpc_service->GetSelectedPort();
|
||||
std::string ep = string::Sprintf("127.0.0.1:%d", port);
|
||||
detail::RPCClient client;
|
||||
|
||||
client.AsyncSendVariable(ep, dev_ctx, scope, NCCL_ID_VARNAME);
|
||||
client.Wait();
|
||||
server_thread.join();
|
||||
auto* ptr = rpc_service.release();
|
||||
delete ptr;
|
||||
}
|
||||
Some files were not shown because too many files have changed in this diff Show More
Loading…
Reference in new issue