[Streaming] Build fix (#10233)

This commit is contained in:
kisuke95
2020-08-26 02:37:21 +08:00
committed by GitHub
parent 7a5af7e744
commit 24a7a8a04d
3 changed files with 8 additions and 0 deletions
+2
View File
@@ -51,6 +51,8 @@ namespace ray {
class CoreWorker;
// If you change this options's definition, you must change the options used in
// other files. Please take a global search and modify them !!!
struct CoreWorkerOptions {
// Callback that must be implemented and provided by the language-specific worker
// frontend to execute tasks and return their results.
+3
View File
@@ -479,6 +479,7 @@ class StreamingWorker {
StreamingWorker(const std::string &store_socket, const std::string &raylet_socket,
int node_manager_port, const gcs::GcsClientOptions &gcs_options)
: test_suite_(nullptr), peer_actor_handle_(nullptr) {
// You must keep it same with `src/ray/core_worker/core_worker.h:CoreWorkerOptions`
CoreWorkerOptions options = {
WorkerType::WORKER, // worker_type
Language::PYTHON, // langauge
@@ -499,6 +500,8 @@ class StreamingWorker {
_7), // task_execution_callback
nullptr, // check_signals
nullptr, // gc_collect
nullptr, // spill_objects
nullptr, // restore_spilled_objects
nullptr, // get_lang_stack
nullptr, // kill_main
true, // ref_counting_enabled
+3
View File
@@ -223,6 +223,7 @@ class StreamingQueueTestBase : public ::testing::TestWithParam<uint64_t> {
}
STREAMING_LOG(INFO) << "Sub process: writer.";
// You must keep it same with `src/ray/core_worker/core_worker.h:CoreWorkerOptions`
CoreWorkerOptions options = {
WorkerType::DRIVER, // worker_type
Language::PYTHON, // langauge
@@ -242,6 +243,8 @@ class StreamingQueueTestBase : public ::testing::TestWithParam<uint64_t> {
nullptr, // task_execution_callback
nullptr, // check_signals
nullptr, // gc_collect
nullptr, // spill_objects
nullptr, // restore_spilled_objects
nullptr, // get_lang_stack
nullptr, // kill_main
true, // ref_counting_enabled