wassname/ray
1
0
Fork 0
mirror of https://github.com/wassname/ray.git synced 2026-08-11 11:24:51 +08:00
Code Issues Packages Projects Releases Wiki Activity

[core] Introduce fetch_local to ray.wait (#12526)

Browse Source
This commit is contained in:
Yi Cheng
2020-12-16 23:44:28 -08:00
committed by GitHub
parent 12231ec2a6
commit 40032541dc
19 changed files with 122 additions and 69 deletions
Download Patch File Download Diff File Expand all files Collapse all files
python/ray/_raylet.pyx
+2 -2
View File
@@ -1004,7 +1004,7 @@ cdef class CoreWorker:
return c_object_id.Binary()
def wait(self, object_refs, int num_returns, int64_t timeout_ms,
TaskID current_task_id):
TaskID current_task_id, c_bool fetch_local):
cdef:
c_vector[CObjectID] wait_ids
c_vector[c_bool] results
@@ -1013,7 +1013,7 @@ cdef class CoreWorker:
wait_ids = ObjectRefsToVector(object_refs)
with nogil:
check_status(CCoreWorkerProcess.GetCoreWorker().Wait(
wait_ids, num_returns, timeout_ms, &results))
wait_ids, num_returns, timeout_ms, &results, fetch_local))
assert len(results) == len(object_refs)