jax
727d1204 - Bump up the shard_count for GPU FFT tests

Commit

1 year ago

Bump up the shard_count for GPU FFT tests They seem to be timing out with ASAN and no sharding. PiperOrigin-RevId: 648301571

References

Author

apaszke

Committer

a-googler

Parents