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

[Placement Group] Placement group remove api part 1 (#10063)

Browse Source 
* Added basic rpc calls.

* fix issues.

* Fix the gcs server not getting request issue.

* In Progress.

* Basic logic done. Tests are required.

* In progress.

* In progress in refactoring context.

* Revert "In progress in refactoring context."

This reverts commit 38236256cf1306c60dd203e75d45ceb4509c8106.

* Working now.

* Python test works.

* Lint.

* Addressed code review.

* Addressed code review.

* Lint.

* Added unit tests.

* Done, but one of unit tests fail

* Addressed code review.

* Addressed the last code review.

* Fix the wrong test case.
This commit is contained in:
SangBin Cho
2020-08-18 12:44:00 -07:00
committed by GitHub
parent d188becec2
commit 263df6163c
24 changed files with 762 additions and 111 deletions
Download Patch File Download Diff File Expand all files Collapse all files
python/ray/experimental/__init__.py
+4 -2
View File
@@ -1,8 +1,10 @@
from .api import get, wait
from .dynamic_resources import set_resource
from .object_spilling import force_spill_objects, force_restore_spilled_objects
from .placement_group import (placement_group, placement_group_table)
from .placement_group import (placement_group, placement_group_table,
remove_placement_group)
__all__ = [
"get", "wait", "set_resource", "force_spill_objects",
"force_restore_spilled_objects", "placement_group", "placement_group_table"
"force_restore_spilled_objects", "placement_group",
"placement_group_table", "remove_placement_group"
]
python/ray/experimental/placement_group.py
+12 -1
View File
@@ -1,6 +1,9 @@
import ray
from typing import (List, Dict)
import ray
from ray._raylet import (
PlacementGroupID, )
def placement_group(bundles: List[Dict[str, float]],
strategy: str = "PACK",
@@ -32,6 +35,14 @@ def placement_group(bundles: List[Dict[str, float]],
return placement_group_id
def remove_placement_group(placement_group_id: PlacementGroupID):
assert type(placement_group_id) == PlacementGroupID
worker = ray.worker.global_worker
worker.check_connected()
worker.core_worker.remove_placement_group(placement_group_id)
def placement_group_table(placement_group_id):
assert placement_group_id is not None
worker = ray.worker.global_worker