legate.core.AutoTask.add_broadcast

AutoTask.add_broadcast(

self,

LogicalStore store,

axes: int | Iterable[int] | None = None,

) → void

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

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

Parameters:

• store (LogicalStore) – LogicalStore 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.