jax
3ebe54ed - [Pallas:MGPU] Increase shard_count for the interpret test

Commit

173 days ago

[Pallas:MGPU] Increase shard_count for the interpret test Otherwise it times out sometimes under ASAN. PiperOrigin-RevId: 862260538

Author

apaszke

Committer

Google-ML-Automation

Parents