Windows changes (#7315)

python/ray/__init__.py

@@ -32,7 +32,8 @@ sys.path.insert(0, thirdparty_files)
 
 # Expose ray ABI symbols which may be dependent by other shared
 # libraries such as _streaming.so. See BUILD.bazel:_raylet
-so_path = os.path.join(dirname(__file__), "_raylet.so")
+python_shared_lib_suffix = ".so" if sys.platform != "win32" else ".pyd"
+so_path = os.path.join(dirname(__file__), "_raylet" + python_shared_lib_suffix)
 if os.path.exists(so_path):
     import ctypes
     from ctypes import CDLL
@@ -158,13 +159,3 @@ __all__ += [
     "TaskID",
     "UniqueID",
 ]
-
-import ctypes  # noqa: E402
-# 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)  # noqa: E501

python/ray/services.py

@@ -6,7 +6,6 @@ import multiprocessing
 import os
 import random
 import re
-import resource
 import socket
 import subprocess
 import sys
@@ -19,6 +18,10 @@ import ray
 import ray.ray_constants as ray_constants
 import psutil
 
+resource = None
+if sys.platform != "win32":
+    import resource
+
 # True if processes are run in the valgrind profiler.
 RUN_RAYLET_PROFILER = False
 RUN_PLASMA_STORE_PROFILER = False
@@ -903,7 +906,7 @@ def _start_redis_instance(executable,
     # number of Redis clients.
     if redis_max_clients is not None:
         redis_client.config_set("maxclients", str(redis_max_clients))
-    else:
+    elif resource is not None:
         # If redis_max_clients is not provided, determine the current ulimit.
         # We will use this to attempt to raise the maximum number of Redis
         # clients.