jax

a9edc15d - Added rocm check in psend lowering method