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

[Serve] Async Router (#6873)

Browse Source
This commit is contained in:
Simon Mo
2020-01-30 09:34:47 -08:00
committed by GitHub
parent 1e3a34b223
commit 660eef6502
15 changed files with 677 additions and 493 deletions
Download Patch File Download Diff File Expand all files Collapse all files
python/ray/async_compat.py
+3 -1
View File
@@ -87,10 +87,12 @@ def get_async(object_id):
if object_id.is_direct_call_type():
inner_future = loop.create_future()
# We must add the done_callback before sending to in_memory_store_get
inner_future.add_done_callback(done_callback)
core_worker.in_memory_store_get_async(object_id, inner_future)
else:
inner_future = as_future(object_id)
inner_future.add_done_callback(done_callback)
inner_future.add_done_callback(done_callback)
# A hack to keep reference to inner_future so it doesn't get GC.
user_future.inner_future = inner_future
# A hack to keep a reference to the object ID for ref counting.