wassname/ray
1
0
Fork 0
mirror of https://github.com/wassname/ray.git synced 2026-07-01 04:50:40 +08:00
Code Issues Packages Projects Releases Wiki Activity

Handle exchange of direct call objects between tasks and actors (#6147)

Browse Source
This commit is contained in:
Eric Liang
2019-11-14 17:32:04 -08:00
committed by GitHub
parent 385783fcec
commit 8ff393a7bd
23 changed files with 426 additions and 202 deletions
Download Patch File Download Diff File Expand all files Collapse all files
python/ray/serialization.py
+2 -6
View File
@@ -158,9 +158,7 @@ class SerializationContext(object):
def id_serializer(obj):
if isinstance(obj, ray.ObjectID) and obj.is_direct_call_type():
raise NotImplementedError(
"Objects produced by direct actor calls cannot be "
"passed to other tasks as arguments.")
obj = self.worker.core_worker.promote_object_to_plasma(obj)
return pickle.dumps(obj)
def id_deserializer(serialized_obj):
@@ -191,9 +189,7 @@ class SerializationContext(object):
def id_serializer(obj):
if isinstance(obj, ray.ObjectID) and obj.is_direct_call_type():
raise NotImplementedError(
"Objects produced by direct actor calls cannot be "
"passed to other tasks as arguments.")
obj = self.worker.core_worker.promote_object_to_plasma(obj)
return obj.__reduce__()
def id_deserializer(serialized_obj):