def loop_body(state):
    effect[0] = True
    (pos, count) = state
    return (lax.add(pos, 1), lax.add(count, inc))
def loop_body(state):
    effect[0] = True
    (pos, count) = state
    return (lax.add(pos, 1), lax.add(count, inc))