wassname/ray
1
0
Fork 0
mirror of https://github.com/wassname/ray.git synced 2026-07-17 11:32:33 +08:00
Code Issues Packages Projects Releases Wiki Activity

Allow tasks to be used with placement groups (#9738)

Browse Source
This commit is contained in:
Eric Liang
2020-07-31 10:51:37 -07:00
committed by GitHub
parent 78995d085f
commit b73080c85f
11 changed files with 177 additions and 120 deletions
Download Patch File Download Diff File Expand all files Collapse all files
python/ray/includes/libcoreworker.pxd
+4 -2
View File
@@ -5,6 +5,7 @@
from libc.stdint cimport int64_t
from libcpp cimport bool as c_bool
from libcpp.memory cimport shared_ptr, unique_ptr
from libcpp.pair cimport pair as c_pair
from libcpp.string cimport string as c_string
from libcpp.unordered_map cimport unordered_map
from libcpp.utility cimport pair
@@ -87,14 +88,15 @@ cdef extern from "ray/core_worker/core_worker.h" nogil:
const CRayFunction &function,
const c_vector[unique_ptr[CTaskArg]] &args,
const CTaskOptions &options, c_vector[CObjectID] *return_ids,
int max_retries)
int max_retries,
c_pair[CPlacementGroupID, int64_t] placement_options)
CRayStatus CreateActor(
const CRayFunction &function,
const c_vector[unique_ptr[CTaskArg]] &args,
const CActorCreationOptions &options,
const c_string &extension_data, CActorID *actor_id)
CRayStatus CreatePlacementGroup(
const CPlacementGroupCreationOptions &options,
const CPlacementGroupCreationOptions &options,
CPlacementGroupID *placement_group_id)
void SubmitActorTask(
const CActorID &actor_id, const CRayFunction &function,