def fun(x):
    ones = jnp.ones((3, 4))
    assert (isinstance(ones, onp.ndarray) and (ones.strides == (0, 0)))
    c = x._node.c
    Broadcast = c.Broadcast
    was_called = []
    c.Broadcast = (lambda *args: (was_called.append(True) or Broadcast(*args)))
    out = (x + ones)
    assert was_called, 'Broadcast was not called.'
    return out
def fun(x):
    ones = jnp.ones((3, 4))
    assert (isinstance(ones, onp.ndarray) and (ones.strides == (0, 0)))
    c = x._node.c
    Broadcast = c.Broadcast
    was_called = []
    c.Broadcast = (lambda *args: (was_called.append(True) or Broadcast(*args)))
    out = (x + ones)
    assert was_called, 'Broadcast was not called.'
    return out