legate.core.operation.AutoTask.add_broadcast

AutoTask.add_broadcast(store: Store, axes: Optional[Union[int, Iterable[int]]] = None) → None

Sets a broadcasting constraint on the store. Equivalent to the following code:

symb = op.declare_partition(store)
op.add_constraint(symb.broadcast(axes))

Parameters:

• store (Store) – Store to set a broadcasting constraint on

• axes (int or Iterable[int], optional) – Axes to broadcast. The entire store is replicated if no axes are given.