def jitted_f(*args):
err, out = f(*args)
checkify.check_error(err)
return out
def f():
checkify.check_error(checkify.Error(True, 0, {0: "hi"}))
def f(x):
_, (xs, errs) = jax.lax.scan(checked_body, (None, ), x)
checkify.check_error(errs)
return xs
def f(pred): # note: data dependence needed!
checkify.check_error(checkify.Error(~pred, 0, {0: "hi"}))
