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.