jax
c28c46e1 - Add ShardedDeviceArray indexing benchmark. (#2549)

Commit
5 years ago
Add ShardedDeviceArray indexing benchmark. (#2549) Example output: ``` ---------Benchmark summary for ShardedDeviceArray_indexing--------- indices_fn mean %std relative ------------------ -------- ------- ---------- integer_indices 0.16901 8.52522 1 integer_2D_indices 18.4918 0 109.412 ```
Author
Parents
Loading