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

Add pending task dependencies to ObjectID ref counting (#6054)

Browse Source
This commit is contained in:
Edward Oakes
2019-11-07 18:37:10 -08:00
committed by GitHub
parent 1f043daf69
commit ca53af4d0f
9 changed files with 395 additions and 67 deletions
Download Patch File Download Diff File Expand all files Collapse all files
python/ray/_raylet.pyx
+4 -4
View File
@@ -1000,17 +1000,17 @@ cdef class CoreWorker:
c_actor_id, &output))
return output
def add_active_object_id(self, ObjectID object_id):
def add_object_id_reference(self, ObjectID object_id):
cdef:
CObjectID c_object_id = object_id.native()
# Note: faster to not release GIL for short-running op.
self.core_worker.get().AddActiveObjectID(c_object_id)
self.core_worker.get().AddObjectIDReference(c_object_id)
def remove_active_object_id(self, ObjectID object_id):
def remove_object_id_reference(self, ObjectID object_id):
cdef:
CObjectID c_object_id = object_id.native()
# Note: faster to not release GIL for short-running op.
self.core_worker.get().RemoveActiveObjectID(c_object_id)
self.core_worker.get().RemoveObjectIDReference(c_object_id)
# TODO: handle noreturn better
cdef store_task_outputs(