legate.core.AutoTask.add_broadcast

AutoTask.add_broadcast(

self,

array_or_store,

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

) → void

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

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

Parameters:

• array_or_store (LogicalArray or LogicalStore) – LogicalArray or LogicalStore to set a broadcasting constraint on

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