Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Remove barrier test in pjrt/distributed with equivalent test coverage…
… in coordination service's fork of client_server_test. 1. This is in preparation for a change in barrier semantics (don't require users to specify unique ids). 2. Moving forward, we want to shift new business logic tests to be in coord service's test suite. This allows us to cover more edge cases with intrusive hooks (e.g. agent dtor tests) as well as non-Jax (i.e. TF) scenarios. 3. Only use pjrt/distributed tests + Jax multi-process Python tests for Jax-specific requirements (e.g. topology exchange contract), or to exercise the xla_client (nanobind) and pjrt/distributed codepaths to validate that args are plumbed correctly. This reduces review burden on the Jax team. PiperOrigin-RevId: 693752761
- Loading branch information