legate.core.AutoTask.add_reduction

AutoTask.add_reduction(self, array_or_store, int32_t redop, partition: Variable | None = None) → Variable

Adds a logical array/store to the task for reduction

Parameters:

• array_or_store (LogicalArray or LogicalStore) – LogicalArray or LogicalStore to pass for reduction

• redop (int) – Reduction operator ID

• partition (Variable, optional) – Partition to associate with the array/store. The default partition is picked if none is given.

Raises:

RuntimeError – If the AutoTask has been previously locked by a PyTask.