mirror of
https://github.com/wassname/ray.git
synced 2026-07-31 12:41:01 +08:00
56 lines
2.3 KiB
Python
56 lines
2.3 KiB
Python
from __future__ import absolute_import
|
|
from __future__ import division
|
|
from __future__ import print_function
|
|
|
|
import os
|
|
import subprocess
|
|
import time
|
|
|
|
|
|
def start_global_scheduler(redis_address, node_ip_address,
|
|
use_valgrind=False, use_profiler=False,
|
|
stdout_file=None, stderr_file=None):
|
|
"""Start a global scheduler process.
|
|
|
|
Args:
|
|
redis_address (str): The address of the Redis instance.
|
|
node_ip_address: The IP address of the node that this scheduler will
|
|
run on.
|
|
use_valgrind (bool): True if the global scheduler should be started
|
|
inside of valgrind. If this is True, use_profiler must be False.
|
|
use_profiler (bool): True if the global scheduler should be started
|
|
inside a profiler. If this is True, use_valgrind must be False.
|
|
stdout_file: A file handle opened for writing to redirect stdout to. If
|
|
no redirection should happen, then this should be None.
|
|
stderr_file: A file handle opened for writing to redirect stderr to. If
|
|
no redirection should happen, then this should be None.
|
|
|
|
Return:
|
|
The process ID of the global scheduler process.
|
|
"""
|
|
if use_valgrind and use_profiler:
|
|
raise Exception("Cannot use valgrind and profiler at the same time.")
|
|
global_scheduler_executable = os.path.join(
|
|
os.path.abspath(os.path.dirname(__file__)),
|
|
"../core/src/global_scheduler/global_scheduler")
|
|
command = [global_scheduler_executable,
|
|
"-r", redis_address,
|
|
"-h", node_ip_address]
|
|
if use_valgrind:
|
|
pid = subprocess.Popen(["valgrind",
|
|
"--track-origins=yes",
|
|
"--leak-check=full",
|
|
"--show-leak-kinds=all",
|
|
"--leak-check-heuristics=stdstring",
|
|
"--error-exitcode=1"] + command,
|
|
stdout=stdout_file, stderr=stderr_file)
|
|
time.sleep(1.0)
|
|
elif use_profiler:
|
|
pid = subprocess.Popen(["valgrind", "--tool=callgrind"] + command,
|
|
stdout=stdout_file, stderr=stderr_file)
|
|
time.sleep(1.0)
|
|
else:
|
|
pid = subprocess.Popen(command, stdout=stdout_file, stderr=stderr_file)
|
|
time.sleep(0.1)
|
|
return pid
|