[Placement Group]Add get all placement group api (#11460)

* add get all interface for placement group

* add get all interface for placement group

* make it work

* fix lint

* fix lint

* fix comment

* add cpp test

* fix python lint
This commit is contained in:
DK.Pino
2020-10-23 11:46:48 -07:00
committed by GitHub
parent e7aa6441b7
commit 9f804ade5f
18 changed files with 157 additions and 19 deletions
+23
View File
@@ -406,6 +406,7 @@ def test_placement_group_table(ray_start_cluster):
# Now the placement group should be scheduled.
cluster.add_node(num_cpus=5, num_gpus=1)
cluster.wait_for_nodes()
actor_1 = Actor.options(
placement_group=placement_group,
@@ -415,6 +416,28 @@ def test_placement_group_table(ray_start_cluster):
result = ray.util.placement_group_table(placement_group)
assert result["state"] == "CREATED"
# Add tow more placement group for placement group table test.
second_strategy = "SPREAD"
ray.util.placement_group(
name="second_placement_group",
strategy=second_strategy,
bundles=bundles)
ray.util.placement_group(
name="third_placement_group",
strategy=second_strategy,
bundles=bundles)
placement_group_table = ray.util.placement_group_table()
assert len(placement_group_table) == 3
true_name_set = {"name", "second_placement_group", "third_placement_group"}
get_name_set = set()
for _, placement_group_data in placement_group_table.items():
get_name_set.add(placement_group_data["name"])
assert true_name_set == get_name_set
def test_cuda_visible_devices(ray_start_cluster):
@ray.remote(num_gpus=1)