jax
c1f01222 - generalize pmap collective-permute tests

Commit

6 years ago

generalize pmap collective-permute tests

References

#690 - add collectives: allreduce min/max, permute

Author

mattjj

mattjj

Parents

Loading