Fix rocm CI

[jainapurva]

Due to the recent version bump to python 3.10, and torch nightlies 2.10.0dev, rocm 6.3 has support errors. Minimum supported version is 7.0

[pytorch-bot]

🔗 Helpful Links

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

• 📄 Preview Python docs built from this PR

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

✅ No Failures

As of commit 9396fc8 with merge base a5a8fe2 ():
💚 Looks good so far! There are no failures yet. 💚

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

[jainapurva]

@jerryzh168 Sorry this fix might not be the best one, as Andrey mentioned that rocm7.0 is not supported by pytorch2.9

[jithunnair-amd]

@jerryzh168 Sorry this fix might not be the best one, as Andrey mentioned that rocm7.0 is not supported by pytorch2.9

Pytorch nightly wheels definitely support ROCm7.0: https://download.pytorch.org/whl/nightly/rocm7.0/

Is there a reason we can't use those?

@jagadish-amd Can you please work with @jainapurva to update the ROCm workflow?

[jagadish-amd]

@jerryzh168 Sorry this fix might not be the best one, as Andrey mentioned that rocm7.0 is not supported by pytorch2.9

Pytorch nightly wheels definitely support ROCm7.0: https://download.pytorch.org/whl/nightly/rocm7.0/

Is there a reason we can't use those?

@jagadish-amd Can you please work with @jainapurva to update the ROCm workflow?

The [build / build-manywheel-py3_10-rocm7_0] job was success earlier with
2025-10-09T22:23:17.7453622Z ++ export 'PIP_INSTALL_TORCH=pip install torch --pre --index-url https://download.pytorch.org/whl/nightly/rocm7.0'
We can verify if the test job passes with below config

    include:
      - name: ROCM Nightly
        runs-on: linux.rocm.gpu.gfx942.2
        torch-spec: '--pre torch --index-url https://download.pytorch.org/whl/nightly/rocm7.0'
        gpu-arch-type: "rocm"
        gpu-arch-version: "7.0"
        docker-image: pytorch/manylinux2_28-builder:rocm7.0

[jainapurva]

@jerryzh168 Sorry this fix might not be the best one, as Andrey mentioned that rocm7.0 is not supported by pytorch2.9

Pytorch nightly wheels definitely support ROCm7.0: https://download.pytorch.org/whl/nightly/rocm7.0/
Is there a reason we can't use those?
@jagadish-amd Can you please work with @jainapurva to update the ROCm workflow?

The [build / build-manywheel-py3_10-rocm7_0] job was success earlier with 2025-10-09T22:23:17.7453622Z ++ export 'PIP_INSTALL_TORCH=pip install torch --pre --index-url https://download.pytorch.org/whl/nightly/rocm7.0' We can verify if the test job passes with below config

    include:
      - name: ROCM Nightly
        runs-on: linux.rocm.gpu.gfx942.2
        torch-spec: '--pre torch --index-url https://download.pytorch.org/whl/nightly/rocm7.0'
        gpu-arch-type: "rocm"
        gpu-arch-version: "7.0"
        docker-image: pytorch/manylinux2_28-builder:rocm7.0

Yes, I tested the flow earlier, please refer to this commit, it was a success. Please let me know if I should just revert the PR to this commit, and remove all extra changes.

[jagadish-amd]

@jerryzh168 Sorry this fix might not be the best one, as Andrey mentioned that rocm7.0 is not supported by pytorch2.9

Pytorch nightly wheels definitely support ROCm7.0: https://download.pytorch.org/whl/nightly/rocm7.0/
Is there a reason we can't use those?
@jagadish-amd Can you please work with @jainapurva to update the ROCm workflow?

The [build / build-manywheel-py3_10-rocm7_0] job was success earlier with 2025-10-09T22:23:17.7453622Z ++ export 'PIP_INSTALL_TORCH=pip install torch --pre --index-url https://download.pytorch.org/whl/nightly/rocm7.0' We can verify if the test job passes with below config

    include:
      - name: ROCM Nightly
        runs-on: linux.rocm.gpu.gfx942.2
        torch-spec: '--pre torch --index-url https://download.pytorch.org/whl/nightly/rocm7.0'
        gpu-arch-type: "rocm"
        gpu-arch-version: "7.0"
        docker-image: pytorch/manylinux2_28-builder:rocm7.0

Yes, I tested the flow earlier, please refer to this commit, it was a success. Please let me know if I should just revert the PR to this commit, and remove all extra changes.

yes, let's go with the commit which has 7.0 , thanks.