mrq/bitsandbytes-rocm
1
0
Fork 1
Code Issues 1 Pull Requests Packages Projects Releases Wiki Activity
2331212b35
bitsandbytes-rocm/speed_benchmark
History
Mitchell Wortsman 2331212b35 add readme for speed bench
2023-04-01 19:13:15 +00:00
..
info_a100_py2.jsonl cleaning and refactor 2023-04-01 18:46:04 +00:00
make_plot_with_jsonl.py cleaning and refactor 2023-04-01 18:46:04 +00:00
plot_with_info.pdf cleaning and refactor 2023-04-01 18:46:04 +00:00
README.md add readme for speed bench 2023-04-01 19:13:15 +00:00
speed_benchmark.py cleaning and refactor 2023-04-01 18:46:04 +00:00

README.md

Steps:

  1. Run python speed_benchmark/speed_benchmark.py which times operations and writes their time to speed_benchmark/info_a100_py2.jsonl (feel free to change the name for your profiling).
  2. Run python speed_benchmark/make_plot_with_jsonl.py, which produces the speed_benchmark/plot_with_info.pdf.