Skip to content

Commit

Permalink
[core][aDAG] Fix microbenchmark regression adag 2 (#47683)
Browse files Browse the repository at this point in the history
After multi ref PR, we cannot just do await on returned value when it is multi ref output
  • Loading branch information
rkooo567 authored Sep 16, 2024
1 parent 2b69277 commit 34a59c6
Showing 1 changed file with 6 additions and 1 deletion.
7 changes: 6 additions & 1 deletion python/ray/_private/ray_experimental_perf.py
Original file line number Diff line number Diff line change
Expand Up @@ -156,7 +156,12 @@ def _exec(dag, num_args=1, payload_size=1):
async def exec_async(tag):
async def _exec_async():
fut = await compiled_dag.execute_async(b"x")
await fut
if not isinstance(fut, list):
await fut
else:
# TODO(sang): Right now, it doesn't work with asyncio.gather.
for f in fut:
await f

return await asyncio_timeit(
tag,
Expand Down

0 comments on commit 34a59c6

Please sign in to comment.