xla
5cb473a2 - [Pallas] Add test_pallas_spmd.py to tpu ci (#7045)

Commit

1 year ago

[Pallas] Add test_pallas_spmd.py to tpu ci (#7045) Summary: Test test_pallas_spmd.py on TPU CI. Test Plan: CI

References

Author

alanwaketan

Parents