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