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

Fix test_task_forward (#5040)

Browse Source
This commit is contained in:
Hao Chen
2019-06-27 14:37:00 +08:00
committed by GitHub
parent 469ae41013
commit a1156754e9
1 changed files with 11 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
python/ray/tests/perf_integration_tests/test_perf_integration.py
+11 -3
View File
@@ -53,9 +53,17 @@ def benchmark_task_forward(f, num_tasks):
"num_tasks", [10**3, 10**4],
ids=[str(num) + "_tasks" for num in [10**3, 10**4]])
def test_task_forward(benchmark, num_tasks):
with _ray_start_cluster(num_cpus=16, object_store_memory=10**6) as cluster:
cluster.add_node(resources={"my_resource": 100})
ray.init(redis_address=cluster.redis_address)
with _ray_start_cluster(
do_init=True,
num_nodes=1,
num_cpus=16,
object_store_memory=10**7,
) as cluster:
cluster.add_node(
num_cpus=16,
object_store_memory=10**7,
resources={"my_resource": 100},
)
@ray.remote(resources={"my_resource": 0.001})
def f():