未验证 提交 ff6a1450 编写于 作者: L lilong12 提交者: GitHub

update, test=develop (#29559)

上级 2cb6f948
...@@ -272,8 +272,7 @@ void GlooWrapper::Init() { ...@@ -272,8 +272,7 @@ void GlooWrapper::Init() {
attr.iface = iface_; attr.iface = iface_;
std::shared_ptr<gloo::rendezvous::HdfsStore> file_store = nullptr; std::shared_ptr<gloo::rendezvous::HdfsStore> file_store = nullptr;
std::shared_ptr<gloo::rendezvous::HTTPStore> http_store = nullptr; std::shared_ptr<gloo::rendezvous::HTTPStore> http_store = nullptr;
auto context = auto context = std::make_shared<gloo::rendezvous::Context>(rank_, size_);
std::make_shared<gloo::rendezvous::ParallelConnectContext>(rank_, size_);
context->setTimeout(run_timeout_); context->setTimeout(run_timeout_);
auto dev = gloo::transport::tcp::CreateDevice(attr); auto dev = gloo::transport::tcp::CreateDevice(attr);
switch (store_type_) { switch (store_type_) {
...@@ -295,6 +294,7 @@ void GlooWrapper::Init() { ...@@ -295,6 +294,7 @@ void GlooWrapper::Init() {
http_store->SetTimeoutSeconds(init_timeout_.count()); http_store->SetTimeoutSeconds(init_timeout_.count());
context->connectFullMesh(*http_store, dev); context->connectFullMesh(*http_store, dev);
http_store->Finalize(); http_store->Finalize();
VLOG(3) << "after calling http_store->Finalize.";
break; break;
} }
default: default:
...@@ -304,6 +304,7 @@ void GlooWrapper::Init() { ...@@ -304,6 +304,7 @@ void GlooWrapper::Init() {
context_ = std::move(context); context_ = std::move(context);
#endif #endif
is_initialized_ = true; is_initialized_ = true;
VLOG(3) << "gloo initialized done.";
} }
template std::vector<int64_t> GlooWrapper::AllReduce<int64_t>( template std::vector<int64_t> GlooWrapper::AllReduce<int64_t>(
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册