Remove test_gather_benchmark (#6983)

This commit is contained in:
Simon Mo
2020-01-31 09:42:05 -08:00
committed by GitHub
parent 64c8996a43
commit 4e2c4302e8
@@ -46,42 +46,6 @@ def test_gather(init):
assert all(a == b for a, b in zip(results, ray.get(tasks)))
def test_gather_benchmark(init):
@ray.remote
def f(n):
time.sleep(0.001 * n)
return 42
async def test_async():
sum_time = 0.
for _ in range(50):
tasks = [f.remote(n) for n in range(20)]
start = time.time()
futures = [async_api.as_future(obj_id) for obj_id in tasks]
await asyncio.gather(*futures)
sum_time += time.time() - start
return sum_time
def baseline():
sum_time = 0.
for _ in range(50):
tasks = [f.remote(n) for n in range(20)]
start = time.time()
ray.get(tasks)
sum_time += time.time() - start
return sum_time
# warm up
baseline()
# async get
sum_time_1 = asyncio.get_event_loop().run_until_complete(test_async())
# get
sum_time_2 = baseline()
# Ensure the new implementation is not too slow.
assert sum_time_2 * 1.2 > sum_time_1
def test_wait(init):
loop = asyncio.get_event_loop()
tasks = gen_tasks()