mirror of
https://github.com/wassname/ray.git
synced 2026-07-23 13:10:11 +08:00
[Object Spilling] Skip normal ray.get path when spilling objects. (#13831)
This commit is contained in:
@@ -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
|
||||
|
||||
Reference in New Issue
Block a user