[Object Spilling] Skip normal ray.get path when spilling objects. (#13831)

This commit is contained in:
SangBin Cho
2021-02-01 16:03:34 -08:00
committed by GitHub
parent e4d30430c0
commit 886217c333
7 changed files with 95 additions and 5 deletions
+12
View File
@@ -898,6 +898,18 @@ cdef class CoreWorker:
return RayObjectsToDataMetadataPairs(results)
def get_if_local(self, object_refs):
"""Get objects from local plasma store directly
without a fetch request to raylet."""
cdef:
c_vector[shared_ptr[CRayObject]] results
c_vector[CObjectID] c_object_ids = ObjectRefsToVector(object_refs)
with nogil:
check_status(
CCoreWorkerProcess.GetCoreWorker().GetIfLocal(
c_object_ids, &results))
return RayObjectsToDataMetadataPairs(results)
def object_exists(self, ObjectRef object_ref):
cdef:
c_bool has_object