wassname/ray
1
0
Fork 0
mirror of https://github.com/wassname/ray.git synced 2026-06-30 10:15:04 +08:00
Code Issues Packages Projects Releases Wiki Activity

[core] Better support multi-nic environments by respecting user-provided IP (#8512)

Browse Source
This commit is contained in:
Xianyang Liu
2020-06-26 03:03:12 +08:00
committed by GitHub
parent 46962f5db1
commit 0bfcc2e5ba
4 changed files with 28 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
python/ray/resource_spec.py
+12 -3
View File
@@ -121,8 +121,14 @@ class ResourceSpec(
return resources
def resolve(self, is_head):
"""Returns a copy with values filled out with system defaults."""
def resolve(self, is_head, node_ip_address=None):
"""Returns a copy with values filled out with system defaults.
Args:
is_head (bool): Whether this is the head node.
node_ip_address (str): The IP address of the node that we are on.
This is used to automatically create a node id resource.
"""
resources = (self.resources or {}).copy()
assert "CPU" not in resources, resources
@@ -130,9 +136,12 @@ class ResourceSpec(
assert "memory" not in resources, resources
assert "object_store_memory" not in resources, resources
if node_ip_address is None:
node_ip_address = ray.services.get_node_ip_address()
# Automatically create a node id resource on each node. This is
# queryable with ray.state.node_ids() and ray.state.current_node_id().
resources[NODE_ID_PREFIX + ray.services.get_node_ip_address()] = 1.0
resources[NODE_ID_PREFIX + node_ip_address] = 1.0
num_cpus = self.num_cpus
if num_cpus is None: