mrq/bitsandbytes-rocm
1
0
Fork 1
Code Issues 1 Pull Requests Packages Projects Releases Wiki Activity

add readme for speed bench

Browse Source
This commit is contained in:
Mitchell Wortsman 2023-04-01 19:13:15 +00:00
parent 7f87ba83ee
commit 2331212b35
1 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
speed_benchmark/README.md Normal file
View File

@ -0,0 +1,4 @@
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`.