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

Add python tests for serialized object ID reference counting (#7038)

Browse Source
This commit is contained in:
Edward Oakes
2020-02-12 16:52:07 -08:00
committed by GitHub
parent d91d3ea936
commit e904711e74
2 changed files with 373 additions and 25 deletions
Download Patch File Download Diff File Expand all files Collapse all files
python/ray/_raylet.pyx
+3 -5
View File
@@ -303,10 +303,8 @@ cdef void prepare_args(
arg_data, string_to_buffer(serialized_arg.metadata))))
else:
args_vector.push_back(
CTaskArg.PassByReference(
(CObjectID.FromBinary(
core_worker.put_serialized_cobject(
serialized_arg)))))
CTaskArg.PassByReference((CObjectID.FromBinary(
core_worker.put_serialized_cobject(serialized_arg)))))
cdef deserialize_args(
const c_vector[shared_ptr[CRayObject]] &c_args,
@@ -1047,7 +1045,7 @@ cdef class CoreWorker:
ref_counts = {}
while it != c_ref_counts.end():
object_id = ObjectID(dereference(it).first.Binary())
object_id = dereference(it).first.Hex()
ref_counts[object_id] = {
"local": dereference(it).second.first,
"submitted": dereference(it).second.second}