wassname/ray
1
0
Fork 0
mirror of https://github.com/wassname/ray.git synced 2026-07-03 15:10:19 +08:00
Code Issues Packages Projects Releases Wiki Activity

Allow large returns from direct actor calls (#6088)

Browse Source
This commit is contained in:
Eric Liang
2019-11-07 21:28:55 -08:00
committed by GitHub
parent ca53af4d0f
commit 4a28306186
26 changed files with 237 additions and 97 deletions
Download Patch File Download Diff File Expand all files Collapse all files
python/ray/includes/task.pxd
+1 -1
View File
@@ -52,7 +52,7 @@ cdef extern from "ray/common/task/task_spec.h" nogil:
c_bool ArgByRef(uint64_t arg_index) const
int ArgIdCount(uint64_t arg_index) const
CObjectID ArgId(uint64_t arg_index, uint64_t id_index) const
CObjectID ReturnId(uint64_t return_index) const
CObjectID ReturnIdForPlasma(uint64_t return_index) const
const uint8_t *ArgData(uint64_t arg_index) const
size_t ArgDataSize(uint64_t arg_index) const
const uint8_t *ArgMetadata(uint64_t arg_index) const
python/ray/includes/task.pxi
+1 -1
View File
@@ -108,7 +108,7 @@ cdef class TaskSpec:
return_id_list = []
for i in range(self.task_spec.get().NumReturns()):
return_id_list.append(
ObjectID(self.task_spec.get().ReturnId(i).Binary()))
ObjectID(self.task_spec.get().ReturnIdForPlasma(i).Binary()))
return return_id_list
def required_resources(self):