xla
4a1588c3 - [FSDPv2] Fix test_fsdp_v2_multi_slice (#7055)

Commit

1 year ago

[FSDPv2] Fix test_fsdp_v2_multi_slice (#7055) Summary: Fix test_fsdp_v2_multi_slice for TPU v2. Test Plan: CI.

References

Author

alanwaketan

Parents