Clean up block/unblock handling of resources in new scheduler (#12963)

This commit is contained in:
Eric Liang
2020-12-18 16:00:54 -08:00
committed by GitHub
parent 3e492a79ec
commit 6ece291f35
12 changed files with 51 additions and 68 deletions
+1 -4
View File
@@ -17,8 +17,7 @@ import ray.ray_constants as ray_constants
from ray.exceptions import RayTaskError
from ray.cluster_utils import Cluster
from ray.test_utils import (wait_for_condition, SignalActor, init_error_pubsub,
get_error_message, Semaphore,
new_scheduler_enabled)
get_error_message, Semaphore)
def test_failed_task(ray_start_regular, error_pubsub):
@@ -633,8 +632,6 @@ def test_export_large_objects(ray_start_regular, error_pubsub):
assert errors[0].type == ray_constants.PICKLING_LARGE_OBJECT_PUSH_ERROR
@pytest.mark.skipif(
new_scheduler_enabled(), reason="Supposed to deadlock, but it doesn't")
def test_warning_all_tasks_blocked(shutdown_only):
ray.init(
num_cpus=1, _system_config={"debug_dump_period_milliseconds": 500})
@@ -6,7 +6,6 @@ from ray.test_utils import (
generate_system_config_map,
wait_for_condition,
wait_for_pid_to_exit,
new_scheduler_enabled,
)
@@ -21,7 +20,6 @@ def increase(x):
return x + 1
@pytest.mark.skipif(new_scheduler_enabled(), reason="notimpl")
@pytest.mark.parametrize(
"ray_start_regular", [
generate_system_config_map(
-3
View File
@@ -8,7 +8,6 @@ import time
import ray
import ray.ray_constants
import ray.test_utils
from ray.test_utils import new_scheduler_enabled
from ray._raylet import GlobalStateAccessor
@@ -217,8 +216,6 @@ def test_load_report(shutdown_only, max_shapes):
global_state_accessor.disconnect()
@pytest.mark.skipif(
new_scheduler_enabled(), reason="requires placement groups")
def test_placement_group_load_report(ray_start_cluster):
cluster = ray_start_cluster
# Add a head node that doesn't have gpu resource.
+1 -1
View File
@@ -167,7 +167,7 @@ def test_dependency_refcounts(ray_start_regular):
check_refcounts({})
@pytest.mark.skipif(new_scheduler_enabled(), reason="hangs")
@pytest.mark.skipif(new_scheduler_enabled(), reason="dynamic res todo")
def test_actor_creation_task(ray_start_regular):
@ray.remote
def large_object():