mirror of
https://github.com/wassname/ray.git
synced 2026-09-11 12:43:20 +08:00
Support multiple core workers in one process (#7623)
This commit is contained in:
@@ -17,7 +17,7 @@ from ray.includes.common cimport (
|
||||
CBuffer,
|
||||
CRayObject
|
||||
)
|
||||
from ray.includes.libcoreworker cimport CCoreWorker
|
||||
from ray.includes.libcoreworker cimport CFiberEvent
|
||||
from ray.includes.unique_ids cimport (
|
||||
CObjectID,
|
||||
CActorID
|
||||
@@ -72,7 +72,7 @@ cdef class ActorID(BaseID):
|
||||
|
||||
cdef class CoreWorker:
|
||||
cdef:
|
||||
unique_ptr[CCoreWorker] core_worker
|
||||
c_bool is_driver
|
||||
object async_thread
|
||||
object async_event_loop
|
||||
object plasma_event_handler
|
||||
@@ -85,6 +85,7 @@ cdef class CoreWorker:
|
||||
cdef store_task_outputs(
|
||||
self, worker, outputs, const c_vector[CObjectID] return_ids,
|
||||
c_vector[shared_ptr[CRayObject]] *returns)
|
||||
cdef yield_current_fiber(self, CFiberEvent &fiber_event)
|
||||
|
||||
cdef class FunctionDescriptor:
|
||||
cdef:
|
||||
|
||||
Reference in New Issue
Block a user