def get_placement_for_any(placement: Union[Collection[Union[Architecture, Device, Task, "TaskID", Any]], Any, None]) \ -> FrozenSet[Device]: if placement is not None: ps = placement if isinstance(placement, Iterable) and not array.is_array(placement) else [placement] return get_placement_for_set(ps) else: return frozenset(get_all_devices())
def __init__(self, devices=None): """ :param devices: The physical devices to use or None to use all physical devices. """ if devices is None: devices = get_all_devices() self._devices = tuple(devices)
def __init__(self, environments: Collection[TaskEnvironment] = None, **kwds): environments = environments or [ TaskEnvironment(placement=[d]) for d in get_all_devices() ] self.environments = environments self.kwds = kwds
def __init__(self, environments: Collection[TaskEnvironment] = None, scheduler_class=task_runtime.Scheduler, **kwds): assert issubclass(scheduler_class, task_runtime.Scheduler) environments = environments or [ TaskEnvironment(placement=[d]) for d in get_all_devices() ] self.environments = environments self.scheduler_class = scheduler_class self.kwds = kwds