hiro/ray
1
0
Fork 0
mirror of https://github.com/vale981/ray synced 2025-03-06 10:31:39 -05:00
Code Issues Projects Releases Packages Wiki Activity Actions

Fix TwoNodeTest.TestActorTaskCrossNodes testcase when enable gcs service (#7416)

Browse source
This commit is contained in:
fangfengbin 2020-03-03 19:37:38 +08:00 • committed by GitHub
parent 584645cc7d
commit f5b1062ed9
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 6 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

7
src/ray/core_worker/core_worker.cc
View file

@ -6,6 +6,7 @@
#include "ray/core_worker/context.h"
#include "ray/core_worker/transport/direct_actor_transport.h"
#include "ray/core_worker/transport/raylet_transport.h"
#include "ray/gcs/gcs_client/service_based_gcs_client.h"
#include "ray/util/util.h"
namespace {
@ -99,7 +100,11 @@ CoreWorker::CoreWorker(const WorkerType worker_type, const Language language,
}
RAY_LOG(INFO) << "Initializing worker " << worker_context_.GetWorkerID();
// Initialize gcs client.
gcs_client_ = std::make_shared<gcs::RedisGcsClient>(gcs_options);
if (getenv("RAY_GCS_SERVICE_ENABLED") != nullptr) {
gcs_client_ = std::make_shared<ray::gcs::ServiceBasedGcsClient>(gcs_options);
} else {
gcs_client_ = std::make_shared<ray::gcs::RedisGcsClient>(gcs_options);
}
RAY_CHECK_OK(gcs_client_->Connect(io_service_));
actor_manager_ = std::unique_ptr<ActorManager>(new ActorManager(gcs_client_->Actors()));