wassname/ray
1
0
Fork 0
mirror of https://github.com/wassname/ray.git synced 2026-07-08 21:18:48 +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/includes/libcoreworker.pxd
+2 -2
View File
@@ -101,8 +101,8 @@ cdef extern from "ray/core_worker/core_worker.h" nogil:
CActorID DeserializeAndRegisterActorHandle(const c_string &bytes)
CRayStatus SerializeActorHandle(const CActorID &actor_id, c_string
*bytes)
void AddActiveObjectID(const CObjectID &object_id)
void RemoveActiveObjectID(const CObjectID &object_id)
void AddObjectIDReference(const CObjectID &object_id)
void RemoveObjectIDReference(const CObjectID &object_id)
CRayStatus SetClientOptions(c_string client_name, int64_t limit)
CRayStatus Put(const CRayObject &object, CObjectID *object_id)
python/ray/includes/unique_ids.pxi
+2 -2
View File
@@ -146,14 +146,14 @@ cdef class ObjectID(BaseID):
# TODO(edoakes): there are dummy object IDs being created in
# includes/task.pxi before the core worker is initialized.
if hasattr(worker, "core_worker"):
worker.core_worker.add_active_object_id(self)
worker.core_worker.add_object_id_reference(self)
self.in_core_worker = True
def __dealloc__(self):
if self.in_core_worker:
try:
worker = ray.worker.global_worker
worker.core_worker.remove_active_object_id(self)
worker.core_worker.remove_object_id_reference(self)
except Exception as e:
# There is a strange error in rllib that causes the above to
# fail. Somehow the global 'ray' variable corresponding to the