How alpa lower control flow jaxpr into XLA HLO?

for example,cond:

>>> from jax import lax
>>>
>>> def func7(arg):
...   return lax.cond(arg >= 0.,
...                   lambda xtrue: xtrue + 3.,
...                   lambda xfalse: xfalse - 3.,
...                   arg)
...
>>> print(make_jaxpr(func7)(5.))
{ lambda ; a:f32[]. let
    b:bool[] = ge a 0.0
    c:i32[] = convert_element_type[new_dtype=int32 weak_type=False] b
    d:f32[] = cond[
      branches=(
        { lambda ; e:f32[]. let f:f32[] = sub e 3.0 in (f,) }
        { lambda ; g:f32[]. let h:f32[] = add g 3.0 in (h,) }
      )
      linear=(False,)
    ] c a
  in (d,) }

In this situation, how alpa deal with cond eqns if we want to lower it into XLA HLO? @zhisbug

[merrymercy]

This is not supported yet, but we are working on supporting it. see also https://github.com/alpa-projects/alpa/issues/400

[merrymercy]

I think the original question is answered. We can close this issue, move control flow related discussion to #400, and move other questions to new issues.