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

[Placement Group] Allow scheduling a task on any bundle (-1, default) (#9885)

Browse Source 
* wip

* wip

* fix tests

* wip

* wip

* wip

* wip

* wip

* add test

* update

* update

* remov debug

* comments
This commit is contained in:
Eric Liang
2020-08-06 00:05:21 -07:00
committed by GitHub
parent 45597e3158
commit 7d4f204aa8
11 changed files with 296 additions and 101 deletions
Download Patch File Download Diff File Expand all files Collapse all files
python/ray/tests/test_placement_group.py
+11
View File
@@ -135,6 +135,17 @@ def test_placement_group_task_resource_ids(ray_start_cluster):
resources = ray.get(o1)
assert len(resources) == 1, resources
assert "CPU_group_" in list(resources.keys())[0], resources
assert "CPU_group_0_" not in list(resources.keys())[0], resources
# Now retry with a bundle index constraint.
o1 = f.options(
placement_group_id=g1, placement_group_bundle_index=0).remote()
resources = ray.get(o1)
assert len(resources) == 2, resources
keys = list(resources.keys())
assert "CPU_group_" in keys[0], resources
assert "CPU_group_" in keys[1], resources
assert "CPU_group_0_" in keys[0] or "CPU_group_0_" in keys[1], resources
def test_placement_group_hang(ray_start_cluster):