[wip] AMD Benchmarks

[jcaip]

You'll need to run the below in order to download the models:

huggingface-cli login

and then to install the requirements:

git clone -b jcaip/amd-benchmarks [email protected]:pytorch/ao.git
cd ao
pip install -r benchmark_requirements.txt
pip install -e . 

Run the benchmarks from the top level ao folder with:

sh ./scripts/prepare.sh
cd torchao/_models/llama
sh benchmarks.sh

[pytorch-bot]

🔗 Helpful Links

🧪 See artifacts and rendered test results at hud.pytorch.org/pr/pytorch/ao/1383

• 📄 Preview Python docs built from this PR

Note: Links to docs will display an error until the docs builds have been completed.

This comment was automatically generated by Dr. CI and updates every 15 minutes.