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.