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"}))