wassname/ray
1
0
Fork 0
mirror of https://github.com/wassname/ray.git synced 2026-09-10 12:38:43 +08:00
Code Issues Packages Projects Releases Wiki Activity

[Core] Use core worker options to initialize (#10467)

Browse Source 
* fix

* fix

* .
This commit is contained in:
kisuke95
2020-09-07 16:36:43 -07:00
committed by GitHub
parent 8db1f16f25
commit b7003839bd
9 changed files with 149 additions and 216 deletions
Download Patch File Download Diff File Expand all files Collapse all files
streaming/src/test/queue_tests_base.h
+16 -30
View File
@@ -232,36 +232,22 @@ 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
raylet_store_socket_names_[0], // store_socket
raylet_socket_names_[0], // raylet_socket
NextJobId(), // job_id
gcs_options_, // gcs_options
true, // enable_logging
"", // log_dir
true, // install_failure_signal_handler
"127.0.0.1", // node_ip_address
node_manager_port_, // node_manager_port
"127.0.0.1", // raylet_ip_address
"queue_tests", // driver_name
"", // stdout_file
"", // stderr_file
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
false, // is_local_mode
1, // num_workers
nullptr, // terminate_asyncio_thread
"", // serialized_job_config
-1, // metrics_agent_port
};
CoreWorkerOptions options;
options.worker_type = WorkerType::DRIVER;
options.language = Language::PYTHON;
options.store_socket = raylet_store_socket_names_[0];
options.raylet_socket = raylet_socket_names_[0];
options.job_id = NextJobId();
options.gcs_options = gcs_options_;
options.enable_logging = true;
options.install_failure_signal_handler = true;
options.node_ip_address = "127.0.0.1";
options.node_manager_port = node_manager_port_;
options.raylet_ip_address = "127.0.0.1";
options.driver_name = "queue_tests";
options.ref_counting_enabled = true;
options.num_workers = 1;
options.metrics_agent_port = -1;
InitShutdownRAII core_worker_raii(CoreWorkerProcess::Initialize,
CoreWorkerProcess::Shutdown, options);