jax
7cb0e601 - Remove the spmd_mode check from OSS JAX since enhanced barrier is switched on for OSS JAX

Commit

1 year ago

Remove the spmd_mode check from OSS JAX since enhanced barrier is switched on for OSS JAX PiperOrigin-RevId: 625763988

References

Author

yashk2810

Committer

a-googler

Parents