legate.core.ManualTask.add_reduction#

ManualTask.add_reduction(
self,
arg: LogicalStore | LogicalStorePartition,
int32_t redop,
projection=None,
bool is_key_partition=False,
) → void#

Adds a reduction store to the task.

Parameters:

  • arg (LogicalStore | LogicalStorePartition) – LogicalStore or LogicalStorePartition to pass as output.

  • redop (int) – A reduction operator ID

  • projection (tuple[SymbolicExpr, ...] | None) – The projection for the partition (if arg is a LogicalStorePartition). If arg is a LogicalStore, then argument is ignored.

  • is_key_partition (bool) – If True, the arg is used as the key partition for mapping this ManualTask. Ignored when the arg is a LogicalStore.

Raises:

TypeError – If arg is neither a LogicalStore or LogicalStorePartition.

On this page