mirror of
https://github.com/wassname/ray.git
synced 2026-06-27 21:38:18 +08:00
Clean up block/unblock handling of resources in new scheduler (#12963)
This commit is contained in:
@@ -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(
|
||||
|
||||
@@ -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.
|
||||
|
||||
@@ -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():
|
||||
|
||||
Reference in New Issue
Block a user