legate.core.AutoTask#

class legate.core.AutoTask#
__init__(*args, **kwargs)#

Attributes

exception_types

tuple[type, ...]

raw_handle

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.