# Wheels generated by http://rocm-ci.amd.com/job/pytorch-manylinux-wheel-builder_rel-7.1/140/

# Wheels built with the following parameters:
# GFX_ARCH: 'gfx908;gfx90a;gfx942;gfx1030;gfx1100;gfx1101;gfx1200;gfx1201;gfx950;gfx1151;gfx1150'
# PYTORCH_BRANCH: 'release/2.7'
# ROCAUTOMATION_BRANCH: 'release/2.7'
# BUILDER_BRANCH: 'rocm7.1'

wget http://compute-artifactory.amd.com/artifactory/compute-pytorch-rocm/compute-rocm-rel-7.1/20/torch-2.7.1+rocm7.1.0.lw.git4b0d5f80-cp312-cp312-linux_x86_64.whl -O torch-2.7.1+rocm7.1.0.lw.git4b0d5f80-cp312-cp312-linux_x86_64.whl
wget http://compute-artifactory.amd.com/artifactory/compute-pytorch-rocm/compute-rocm-rel-7.1/20/apex-1.7.0a0+rocm7.1.0.git20a62dc5-cp312-cp312-linux_x86_64.whl -O apex-1.7.0a0+rocm7.1.0.git20a62dc5-cp312-cp312-linux_x86_64.whl
wget http://compute-artifactory.amd.com/artifactory/compute-pytorch-rocm/compute-rocm-rel-7.1/20/torchaudio-2.7.1+rocm7.1.0.git95c61b41-cp312-cp312-linux_x86_64.whl -O torchaudio-2.7.1+rocm7.1.0.git95c61b41-cp312-cp312-linux_x86_64.whl
wget http://compute-artifactory.amd.com/artifactory/compute-pytorch-rocm/compute-rocm-rel-7.1/20/torchvision-0.22.1+rocm7.1.0.git59a3e1f9-cp312-cp312-linux_x86_64.whl -O torchvision-0.22.1+rocm7.1.0.git59a3e1f9-cp312-cp312-linux_x86_64.whl
wget http://compute-artifactory.amd.com/artifactory/compute-pytorch-rocm/compute-rocm-rel-7.1/20/triton-3.3.1+rocm7.1.0.git9c7bc0a3-cp312-cp312-linux_x86_64.whl -O triton-3.3.1+rocm7.1.0.git9c7bc0a3-cp312-cp312-linux_x86_64.whl

pip3 install \
  torch-2.7.1+rocm7.1.0.lw.git4b0d5f80-cp312-cp312-linux_x86_64.whl \
  apex-1.7.0a0+rocm7.1.0.git20a62dc5-cp312-cp312-linux_x86_64.whl \
  torchaudio-2.7.1+rocm7.1.0.git95c61b41-cp312-cp312-linux_x86_64.whl \
  torchvision-0.22.1+rocm7.1.0.git59a3e1f9-cp312-cp312-linux_x86_64.whl \
  triton-3.3.1+rocm7.1.0.git9c7bc0a3-cp312-cp312-linux_x86_64.whl
