mirror of
https://github.com/wassname/ray.git
synced 2026-07-01 10:18:53 +08:00
Robert Nishihara
8d90c9f432
* Experimental utils for copying directories to other machines in the cluster using Ray. * Test copying directory functionality. * Small fix.
21 lines
1.2 KiB
Python
21 lines
1.2 KiB
Python
from __future__ import absolute_import
|
|
from __future__ import division
|
|
from __future__ import print_function
|
|
|
|
# Ray version string
|
|
__version__ = "0.01"
|
|
|
|
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 ray.experimental
|
|
import ray.serialization
|
|
from ray.worker import register_class, error_info, init, connect, disconnect, get, put, wait, remote
|
|
from ray.worker import Reusable, reusables
|
|
from ray.worker import SCRIPT_MODE, WORKER_MODE, PYTHON_MODE, SILENT_MODE
|