legate.core.AutoTask#
- class legate.core.AutoTask#
- __init__(*args, **kwargs)#
Attributes
tuple[type, ...]
uintptr_t
Methods
add_alignment(self, array_or_store1, ...)Sets an alignment between stores.
add_broadcast(self, array_or_store[, axes])Sets a broadcasting constraint on the logical_array.
add_cal_communicator(self)Adds a CAL communicator to the task
add_communicator(self, str name)Add a communicator to the task
add_constraint(self, Constraint constraint)Add a partitioning constraint to the task.
add_cpu_communicator(self)Adds a CPU communicator to the task
add_input(self, array_or_store[, partition])Adds a logical array/store as input to the task
add_nccl_communicator(self)Adds a NCCL communicator to the task
add_output(self, array_or_store[, partition])Adds a logical array/store as output to the task
add_reduction(self, array_or_store, ...)Adds a logical array/store to the task for reduction
add_scalar_arg(self, value[, dtype])Adds a by-value argument to the task
declare_partition(self)Declare a partition symbol for this task.
execute(self)Submits the operation to the runtime.
find_or_declare_partition(self, ...)Finds or creates a partition symbol for the given array.
lock(self)Lock an AutoTask from further argument modifications.
provenance(self)Returns the provenance of this task.
set_concurrent(self, bool concurrent)Set whether a task requires a concurrent task launch.
set_side_effect(self, bool has_side_effect)Set whether a task has side effects.
throws_exception(self, type exception_type)Set which exception is thrown by the task.