def _realize_and_check(f, offset=0):
    b = hl.Buffer(hl.Float(32), [2, 2])
    f.realize(b)
    assert (b[(0, 0)] == ((3.5 + offset) + 123))
    assert (b[(0, 1)] == ((4.5 + offset) + 123))
    assert (b[(1, 0)] == ((4.5 + offset) + 123))
    assert (b[(1, 1)] == ((5.5 + offset) + 123))
def _realize_and_check(f, offset=0):
    b = hl.Buffer(hl.Float(32), [2, 2])
    f.realize(b)
    assert (b[(0, 0)] == ((3.5 + offset) + 123))
    assert (b[(0, 1)] == ((4.5 + offset) + 123))
    assert (b[(1, 0)] == ((4.5 + offset) + 123))
    assert (b[(1, 1)] == ((5.5 + offset) + 123))