mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-16 20:06:05 +00:00
Code Issues Packages Projects Releases Wiki Activity
rocm_jax/benchmarks
History
Skye Wanderman-Milne 8c2901cf4a
Add --export_dir and --baseline_dir flags to benchmark.py. (#2677) 
`--export_dir` allows saving benchmark results to CSV files, and
`--baseline_dir` allows comparing results to a baseline exported via
`--export_dir`.
2020-04-13 10:07:05 -07:00
..
benchmark.py
Add --export_dir and --baseline_dir flags to benchmark.py. (#2677)
2020-04-13 10:07:05 -07:00
pmap_benchmark.py
Adjust pmap_bechmark.py values to be more realistic. (#2622)
2020-04-06 16:38:34 -07:00