| 1 | # Common dependencies |
| 2 | -r ../common.txt |
| 3 | |
| 4 | |
| 5 | --extra-index-url https://repo.amd.com/rocm/whl-multi-arch/ |
| 6 | torch==2.12.0+rocm7.14.0 |
| 7 | torchvision==0.27.0+rocm7.14.0 |
| 8 | torchaudio==2.11.0+rocm7.14.0 |
| 9 | triton==3.7.1+git0263a6a6.rocm7.14.0 |
| 10 | cmake>=3.26.1,<4 |
| 11 | packaging>=26.2 |
| 12 | setuptools>=78.1.1,<80.0.0 |
| 13 | setuptools-scm>=8 |
| 14 | setuptools-rust>=1.9.0 |
| 15 | wheel |
| 16 | jinja2>=3.1.6 |
| 17 | amdsmi==7.0.2 |
| 18 | timm>=1.0.17 |
| 19 | tilelang==0.1.10 |