mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-16 11:56:07 +00:00
Code Issues Packages Projects Releases Wiki Activity
rocm_jax/benchmarks
History
Skye Wanderman-Milne c28c46e191
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
```
2020-03-31 15:52:41 -07:00
..
benchmark.py
Add ShardedDeviceArray indexing benchmark. (#2549)
2020-03-31 15:52:41 -07:00
pmap_benchmark.py
Add ShardedDeviceArray indexing benchmark. (#2549)
2020-03-31 15:52:41 -07:00