mirror of
https://github.com/wassname/ray.git
synced 2026-07-12 15:57:53 +08:00
mehrdadn
23 lines
1.3 KiB
Python
23 lines
1.3 KiB
Python
# These three constants are used to define the mode that a worker is running in.
|
|
# Right now, this is only used for determining how to print information about
|
|
# task failures.
|
|
SCRIPT_MODE = 0
|
|
WORKER_MODE = 1
|
|
SHELL_MODE = 2
|
|
PYTHON_MODE = 3
|
|
|
|
import ctypes
|
|
# Windows only
|
|
if hasattr(ctypes, "windll"):
|
|
# Makes sure that all child processes die when we die
|
|
# Also makes sure that fatal crashes result in process termination rather than an error dialog (the latter is annoying since we have a lot of processes)
|
|
# This is done by associating all child processes with a "job" object that imposes this behavior.
|
|
(lambda kernel32: (lambda job: (lambda n: kernel32.SetInformationJobObject(job, 9, "\0" * 17 + chr(0x8 | 0x4 | 0x20) + "\0" * (n - 18), n))(0x90 if ctypes.sizeof(ctypes.c_void_p) > ctypes.sizeof(ctypes.c_int) else 0x70) and kernel32.AssignProcessToJobObject(job, ctypes.c_void_p(kernel32.GetCurrentProcess())))(ctypes.c_void_p(kernel32.CreateJobObjectW(None, None))) if kernel32 is not None else None)(ctypes.windll.kernel32)
|
|
|
|
import config
|
|
import libraylib as lib
|
|
import serialization
|
|
from worker import scheduler_info, visualize_computation_graph, task_info, register_module, connect, disconnect, get, put, remote, kill_workers, restart_workers_local
|
|
from libraylib import ObjRef
|
|
import internal
|