jax

bb96dc60 - Implement warpgroup_barrier() with s_waitcnt + s_barrier